Skip to content

Commit 6a75470

Browse files
fix q6 test
1 parent 060a66b commit 6a75470

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/quantization/ggml/test_ggml.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -691,7 +691,7 @@ def test_starcoder2_q6_k(self):
691691
text = tokenizer(example_function_text, return_tensors="pt").to(torch_device)
692692
out = model.generate(**text, max_new_tokens=10)
693693

694-
EXPECTED_TEXT = 'def print_hello_world():\n\tprint("Hello World")\n\ndef print'
694+
EXPECTED_TEXT = 'def print_hello_world():\n print("Hello World")\n\ndef print'
695695
self.assertEqual(tokenizer.decode(out[0], skip_special_tokens=True), EXPECTED_TEXT)
696696

697697
def test_tokenization_xnli(self):

0 commit comments

Comments
 (0)