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

Revert integration tests modifications.

Browse Source
This commit is contained in:
Nicolas Patry 2023-12-06 09:49:54 +00:00
parent 6350c11df3
commit b3c1492be1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
integration-tests/models/test_flash_falcon.py
View File

@ -54,7 +54,7 @@ async def test_flash_falcon_all_params(flash_falcon, response_snapshot):
async def test_flash_falcon_load(flash_falcon, generate_load, response_snapshot): async def test_flash_falcon_load(flash_falcon, generate_load, response_snapshot):
responses = await generate_load( responses = await generate_load(
flash_falcon, flash_falcon,
"What is Deep Learning?", "Girafatron is obsessed with giraffes, the most glorious animal on the face of this Earth. Giraftron believes all other animals are irrelevant when compared to the glorious majesty of the giraffe.\nDaniel: Hello, Girafatron!\nGirafatron:",
max_new_tokens=10, max_new_tokens=10,
n=4, n=4,
) )