Remove trust_remote_code

huggingface · Dec 5, 2024 · 14c186f · 14c186f
1 parent 02e6dc9
commit 14c186f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/tests/openvino/test_exporters_cli.py b/tests/openvino/test_exporters_cli.py
@@ -181,7 +181,7 @@ class OVCLIExportTestCase(unittest.TestCase):
                     "image-text-to-text",
                     "qwen2_vl",
                     'int4 --group-size 4 --ratio 0.9 --sensitivity-metric "mean_activation_magnitude" '
-                    "--dataset contextual --num-samples 1 --trust-remote-code",
+                    "--dataset contextual --num-samples 1",
                     {"int8": 8, "int4": 22},
                 ),
             ]

diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py
@@ -418,15 +418,14 @@ class OVWeightCompressionTest(unittest.TestCase):
                 (
                     OVModelForVisualCausalLM,
                     "qwen2_vl",
-                    True,
+                    False,
                     dict(
                         bits=4,
                         group_size=16,
                         dataset="contextual",
                         ratio=0.8,
                         sensitivity_metric="mean_activation_magnitude",
                         num_samples=1,
-                        trust_remote_code=True,
                     ),
                     {"int4": 20, "int8": 10},
                 ),