Needed to regenerate params tests + fix simple tests

huggingface · Dec 4, 2023 · de6594a · de6594a
1 parent 32878de
commit de6594a
Show file tree

Hide file tree

Showing 11 changed files with 174 additions and 241 deletions.
diff --git a/...on-tests/models/__snapshots__/test_flash_llama_gptq/test_flash_llama_gptq_all_params.json b/...on-tests/models/__snapshots__/test_flash_llama_gptq/test_flash_llama_gptq_all_params.json
@@ -11,78 +11,78 @@
       },
       {
         "id": 4321,
-        "logprob": -9.6015625,
+        "logprob": -9.5625,
         "text": "Test"
       },
       {
         "id": 2009,
-        "logprob": -9.6640625,
+        "logprob": -9.6796875,
         "text": "request"
       }
     ],
     "seed": 0,
     "tokens": [
       {
         "id": 29899,
-        "logprob": -1.1640625,
+        "logprob": -1.1972656,
         "special": false,
         "text": "-"
       },
       {
-        "id": 1454,
-        "logprob": -0.07543945,
+        "id": 29896,
+        "logprob": -1.1621094,
         "special": false,
-        "text": "for"
+        "text": "1"
       },
       {
-        "id": 29899,
-        "logprob": 0.0,
+        "id": 13,
+        "logprob": -0.12768555,
         "special": false,
-        "text": "-"
+        "text": "\n"
       },
       {
-        "id": 9342,
-        "logprob": 0.0,
+        "id": 3057,
+        "logprob": -0.074279785,
         "special": false,
-        "text": "comment"
+        "text": "Test"
       },
       {
-        "id": 29901,
+        "id": 2009,
         "logprob": 0.0,
         "special": false,
-        "text": ":"
+        "text": " request"
       },
       {
-        "id": 396,
-        "logprob": -0.2956543,
+        "id": 29899,
+        "logprob": 0.0,
         "special": false,
-        "text": " #"
+        "text": "-"
       },
       {
         "id": 29906,
-        "logprob": -0.52734375,
+        "logprob": -0.20141602,
         "special": false,
         "text": "2"
       },
       {
-        "id": 29900,
-        "logprob": -0.6899414,
+        "id": 13,
+        "logprob": 0.0,
         "special": false,
-        "text": "0"
+        "text": "\n"
       },
       {
-        "id": 29896,
-        "logprob": 0.0,
+        "id": 3057,
+        "logprob": -0.6611328,
         "special": false,
-        "text": "1"
+        "text": "Test"
       },
       {
-        "id": 29946,
-        "logprob": -1.5068359,
+        "id": 2009,
+        "logprob": 0.0,
         "special": false,
-        "text": "4"
+        "text": " request"
       }
     ]
   },
-  "generated_text": "Test request-for-comment: #2014"
+  "generated_text": "Test request-1\nTest request-2\nTest request"
 }
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
@@ -91,14 +91,8 @@
         "logprob": 0.0,
         "special": false,
         "text": " a"
-      },
-      {
-        "id": 11306,
-        "logprob": -0.5488281,
-        "special": false,
-        "text": " subset"
       }
     ]
   },
-  "generated_text": "What is Deep Learning?\nDeep learning can be thought of as a subset"
+  "generated_text": "What is Deep Learning?\nDeep learning can be thought of as a"
 }