saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show stats 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):
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"