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

fix: small test tweak

Browse Source
This commit is contained in:
drbh 2024-05-14 03:16:01 +00:00 committed by Nicolas Patry
parent 5b3b8fd7b6
commit 9b9614cea3
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
integration-tests/models/test_flash_pali_gemma.py
View File

@ -34,6 +34,4 @@ async def test_flash_pali_gemma(flash_pali_gemma, response_snapshot):
inputs = f"Where is the cow standing?\n![]({cow})" inputs = f"Where is the cow standing?\n![]({cow})"
response = await flash_pali_gemma.generate(inputs, max_new_tokens=20) response = await flash_pali_gemma.generate(inputs, max_new_tokens=20)
# TODO: update this! this is incorrect and just to show the current state of the test assert response.generated_text == "\nbeach"
assert response.generated_text == " - HDS"
# assert response.generated_text == "\nbeach"