Update references

huggingface · Dec 10, 2024 · f3508f4 · f3508f4
1 parent 27358a2
commit f3508f4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/openvino/test_exporters_cli.py b/tests/openvino/test_exporters_cli.py
@@ -148,7 +148,7 @@ class OVCLIExportTestCase(unittest.TestCase):
             "text-generation-with-past",
             "llama_awq",
             "int4 --group-size 16 --backup-precision none --ratio 0.5",
-            [{"int4": 12}],
+            [{"int4": 6}],
         ),
     ]
 

diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py
@@ -382,21 +382,21 @@ class OVWeightCompressionTest(unittest.TestCase):
             "llama_awq",
             False,
             dict(bits=4, backup_precision="none", group_size=16, ratio=0.5),
-            [{"int4": 12}],
+            [{"int4": 6}],
         ),
         (
             OVModelForCausalLM,
             "llama_awq",
             False,
             dict(bits=4, backup_precision="int8_sym", group_size=16, ratio=0.5),
-            [{"int4": 12, "int8": 10}],
+            [{"int4": 6, "int8": 13}],
         ),
         (
             OVModelForCausalLM,
             "llama_awq",
             False,
             dict(bits=4, backup_precision="int8_asym", group_size=16, ratio=0.5),
-            [{"int4": 12, "int8": 20}],
+            [{"int4": 6, "int8": 26}],
         ),
     ]