SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix quanto tests (#31062)

Browse Source 
fix quanto tests
This commit is contained in:
Marc Sun
2024-05-27 15:53:45 +02:00
committed by GitHub
parent cd797778e4
commit b84cd67526
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/quantization/quanto_integration/test_quanto.py
View File

@@ -413,18 +413,18 @@ class QuantoQuantizationSerializationCudaTest(QuantoQuantizationTest):
class QuantoQuantizationQBitsTensorTest(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" weights = "int4"
class QuantoQuantizationQBitsTensorOffloadTest(QuantoQuantizationOffloadTest): 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" weights = "int4"
@unittest.skip("Skipping test class because serialization is not supported yet") @unittest.skip("Skipping test class because serialization is not supported yet")
class QuantoQuantizationQBitsTensorSerializationTest(QuantoQuantizationSerializationTest): 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" weights = "int4"