From b84cd67526eb8bac5bfba47bac72003bacc0eb5a Mon Sep 17 00:00:00 2001 From: Marc Sun <57196510+SunMarc@users.noreply.github.com> Date: Mon, 27 May 2024 15:53:45 +0200 Subject: [PATCH] Fix quanto tests (#31062) fix quanto tests --- tests/quantization/quanto_integration/test_quanto.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/quantization/quanto_integration/test_quanto.py b/tests/quantization/quanto_integration/test_quanto.py index e662300a4..b5020f803 100644 --- a/tests/quantization/quanto_integration/test_quanto.py +++ b/tests/quantization/quanto_integration/test_quanto.py @@ -413,18 +413,18 @@ class QuantoQuantizationSerializationCudaTest(QuantoQuantizationTest): class QuantoQuantizationQBitsTensorTest(QuantoQuantizationTest): - EXPECTED_OUTPUTS = "Hello my name is John, I am a young man from the Philippines" + EXPECTED_OUTPUTS = "Hello my name is Nils, I am a student of the University" weights = "int4" class QuantoQuantizationQBitsTensorOffloadTest(QuantoQuantizationOffloadTest): - EXPECTED_OUTPUTS = "Hello my name is John, I am a young man from the Philippines" + EXPECTED_OUTPUTS = "Hello my name is Nils, I am a student of the University" weights = "int4" @unittest.skip("Skipping test class because serialization is not supported yet") class QuantoQuantizationQBitsTensorSerializationTest(QuantoQuantizationSerializationTest): - EXPECTED_OUTPUTS = "Hello my name is John, I am a young man from the Philippines" + EXPECTED_OUTPUTS = "Hello my name is Nils, I am a student of the University" weights = "int4"