fix test

huggingface · Dec 25, 2023 · 1ccecc3 · 1ccecc3
1 parent a6202a0
commit 1ccecc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/quantization/autoawq/test_awq.py b/tests/quantization/autoawq/test_awq.py
@@ -240,7 +240,7 @@ def test_quantized_model_no_k_proj_quantized(self):
         )
 
         output = quantized_model.generate(dummy_input, max_new_tokens=10)
-        self.assertEqual(EXPECTED_OUTPUT, output)
+        self.assertTrue((EXPECTED_OUTPUT == output).all())
 
 
 @slow