From abc8d48d96f94e4eec0d72f4a66f9298b16a1d2e Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Wed, 6 Dec 2023 19:51:27 +0000 Subject: [PATCH] Old llama test. --- integration-tests/models/test_flash_llama.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integration-tests/models/test_flash_llama.py b/integration-tests/models/test_flash_llama.py index a9536bc4..c69314ff 100644 --- a/integration-tests/models/test_flash_llama.py +++ b/integration-tests/models/test_flash_llama.py @@ -43,7 +43,7 @@ async def test_flash_llama_all_params(flash_llama, response_snapshot): seed=0, ) - assert response.details.generated_tokens == 10 + assert response.details.generated_tokens == 5 assert response == response_snapshot