From 65bc0aaa58dba85ced3d671cf5df491fda3375bf Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Wed, 15 May 2024 10:17:07 +0000 Subject: [PATCH] Working integration-tests. --- .../test_flash_pali_gemma/test_flash_pali_gemma.json | 4 ++-- integration-tests/models/test_flash_pali_gemma.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/integration-tests/models/__snapshots__/test_flash_pali_gemma/test_flash_pali_gemma.json b/integration-tests/models/__snapshots__/test_flash_pali_gemma/test_flash_pali_gemma.json index f78428fb..037e0b16 100644 --- a/integration-tests/models/__snapshots__/test_flash_pali_gemma/test_flash_pali_gemma.json +++ b/integration-tests/models/__snapshots__/test_flash_pali_gemma/test_flash_pali_gemma.json @@ -8,13 +8,13 @@ "tokens": [ { "id": 54901, - "logprob": -0.61621094, + "logprob": -0.72753906, "special": false, "text": "beach" }, { "id": 1, - "logprob": -0.11273193, + "logprob": -0.011009216, "special": true, "text": "" } diff --git a/integration-tests/models/test_flash_pali_gemma.py b/integration-tests/models/test_flash_pali_gemma.py index 61d816e1..d4e83c9f 100644 --- a/integration-tests/models/test_flash_pali_gemma.py +++ b/integration-tests/models/test_flash_pali_gemma.py @@ -7,8 +7,9 @@ import base64 @pytest.fixture(scope="module") def flash_pali_gemma_handle(launcher): with launcher( - "gv-hf/paligemma-3b-mix-224", + "google/paligemma-3b-pt-224", num_shard=1, + revision="float16", max_input_length=4000, max_total_tokens=4096, ) as handle: