From b6519b52797059bce2b9d1e77a86c9aec799fd83 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Sat, 9 Dec 2023 17:08:21 +0000
Subject: [PATCH] Update medusa sampling.

---
 .../test_flash_medusa_all_params.json         | 34 +++++++++----------
 1 file changed, 17 insertions(+), 17 deletions(-)

diff --git a/integration-tests/models/__snapshots__/test_flash_medusa/test_flash_medusa_all_params.json b/integration-tests/models/__snapshots__/test_flash_medusa/test_flash_medusa_all_params.json
index b2068446..d8a298eb 100644
--- a/integration-tests/models/__snapshots__/test_flash_medusa/test_flash_medusa_all_params.json
+++ b/integration-tests/models/__snapshots__/test_flash_medusa/test_flash_medusa_all_params.json
@@ -57,42 +57,42 @@
         "text": " learning"
       },
       {
-        "id": 756,
-        "logprob": -0.48095703,
+        "id": 29892,
+        "logprob": -0.61816406,
         "special": false,
-        "text": " has"
+        "text": ","
       },
       {
-        "id": 19479,
-        "logprob": 0.0,
+        "id": 607,
+        "logprob": -0.7089844,
         "special": false,
-        "text": " revolution"
+        "text": " which"
       },
       {
-        "id": 1891,
-        "logprob": 0.0,
+        "id": 508,
+        "logprob": -1.7724609,
         "special": false,
-        "text": "ized"
+        "text": " can"
       },
       {
-        "id": 278,
+        "id": 367,
         "logprob": 0.0,
         "special": false,
-        "text": " the"
+        "text": " be"
       },
       {
-        "id": 1746,
+        "id": 5545,
         "logprob": 0.0,
         "special": false,
-        "text": " field"
+        "text": " considered"
       },
       {
-        "id": 310,
-        "logprob": 0.0,
+        "id": 408,
+        "logprob": -0.3869629,
         "special": false,
-        "text": " of"
+        "text": " as"
       }
     ]
   },
-  "generated_text": "What is Deep Learning?\nDeep learning has revolutionized the field of"
+  "generated_text": "What is Deep Learning?\nDeep learning, which can be considered as"
 }