huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix URL.

Browse Source
This commit is contained in:
Nicolas Patry 2024-03-21 19:40:52 +00:00
parent 6f18cf7fb4
commit 472bf098e4
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
integration-tests/models/test_idefics.py
View File

@ -15,17 +15,18 @@ async def idefics(idefics_handle):
await idefics_handle.health(300)
return idefics_handle.client
# TODO fix the server parsser to count inline image tokens correctly
def get_chicken():
with open("integrations-tests/images/chicken_on_money.png", "rb") as image_file:
with open("integration-tests/images/chicken_on_money.png", "rb") as image_file:
encoded_string = base64.b64encode(image_file.read())
return f"data:image/png;base64,{encoded_string}"
@pytest.mark.asyncio
async def test_idefics(idefics, response_snapshot):
chicken = get_chicken()
response = await idefics.generate(
f"User:![]({chicken})Can you tell me a very short story based on the image?",
"User:![](https://huggingface.co/spaces/HuggingFaceM4/idefics_playground/resolve/main/example_images/chicken_on_money.png?download=true)Can you tell me a very short story based on the image?",
max_new_tokens=10,
decoder_input_details=True,
)
@ -38,7 +39,7 @@ async def test_idefics(idefics, response_snapshot):
async def test_idefics_load(idefics, generate_load, response_snapshot):
responses = await generate_load(
idefics,
f"User:![]({chicken})Can you tell me a very short story based on the image?",
"User:![](https://huggingface.co/spaces/HuggingFaceM4/idefics_playground/resolve/main/example_images/chicken_on_money.png?download=true)Can you tell me a very short story based on the image?",
max_new_tokens=10,
n=4,
)