Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "b074461ef0f54ce37c5239d30ee960ece28d11ec"
Unverified Commit b84cd675 authored by Marc Sun's avatar Marc Sun Committed by GitHub
Browse files

Fix quanto tests (#31062) 

fix quanto tests
parent cd797778
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/quantization/quanto_integration/test_quanto.py
View file @ b84cd675
...@@ -413,18 +413,18 @@ class QuantoQuantizationSerializationCudaTest(QuantoQuantizationTest): ...@@ -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"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment