more expected putputs with each run

huggingface · May 28, 2024 · 4e75ea1 · 4e75ea1
1 parent 8ca98cb
commit 4e75ea1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/gptq/test_quantization.py b/tests/gptq/test_quantization.py
@@ -54,6 +54,8 @@ class GPTQTest(unittest.TestCase):
     EXPECTED_OUTPUTS.add("Hello my name is nancy and i am a very sweet and very")
     EXPECTED_OUTPUTS.add("Hello my name is nancy, I am a very pretty girl with")
     EXPECTED_OUTPUTS.add("Hello my name is nancy and i am a very pretty girl.")
+    EXPECTED_OUTPUTS.add("Hello my name is jessica and i am a very sweet and")
+    EXPECTED_OUTPUTS.add("Hello my name is Alyson and I am a very sweet and")
     EXPECTED_OUTPUTS.add("Hello my name is jessie and i am a new member of")
     EXPECTED_OUTPUTS.add("Hello my name is A.I. and I am a student of")
     EXPECTED_OUTPUTS.add("Hello my name is A.I.I.I.I.")