fix q6 test

huggingface · Oct 11, 2024 · 11e4b9c · 11e4b9c
1 parent 50cbe4f
commit 11e4b9c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/quantization/ggml/test_ggml.py b/tests/quantization/ggml/test_ggml.py
@@ -691,7 +691,7 @@ def test_starcoder2_q6_k(self):
         text = tokenizer(example_function_text, return_tensors="pt").to(torch_device)
         out = model.generate(**text, max_new_tokens=10)
 
-        EXPECTED_TEXT = 'def print_hello_world():\n\tprint("Hello World")\n\ndef print'
+        EXPECTED_TEXT = 'def print_hello_world():\n    print("Hello World")\n\ndef print'
         self.assertEqual(tokenizer.decode(out[0], skip_special_tokens=True), EXPECTED_TEXT)
 
     def test_tokenization_xnli(self):