diff --git a/integration-tests/models/__snapshots__/test_idefics2/test_flash_idefics2_next_all_params.json b/integration-tests/models/__snapshots__/test_idefics2/test_flash_idefics2_next_all_params.json
index dab437b9..1fad0b96 100644
--- a/integration-tests/models/__snapshots__/test_idefics2/test_flash_idefics2_next_all_params.json
+++ b/integration-tests/models/__snapshots__/test_idefics2/test_flash_idefics2_next_all_params.json
@@ -30,19 +30,19 @@
       },
       {
         "id": 264,
-        "logprob": -0.38061523,
+        "logprob": -0.37573242,
         "special": false,
         "text": " a"
       },
       {
         "id": 633,
-        "logprob": -0.09301758,
+        "logprob": -0.09161377,
         "special": false,
         "text": " new"
       },
       {
         "id": 4480,
-        "logprob": -0.26782227,
+        "logprob": -0.26171875,
         "special": false,
         "text": " feature"
       },
@@ -78,7 +78,7 @@
       },
       {
         "id": 13,
-        "logprob": -0.10632324,
+        "logprob": 0.0,
         "special": false,
         "text": "\n"
       }