fix

echarlaix · echarlaix · commit 206177840230 · 2024-01-30T15:03:29.000+01:00
diff --git a/tests/ipex/test_modeling.py b/tests/ipex/test_modeling.py
@@ -199,7 +199,7 @@ def test_pipeline(self, model_arch):
 class IPEXModelForCausalLMTest(unittest.TestCase):
     SUPPORTED_ARCHITECTURES = (
         "bart",
-        "gpt_bigcode",
+        # "gpt_bigcode",
         "blenderbot",
         "blenderbot-small",
         "bloom",
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -472,7 +472,7 @@ def test_pipeline(self, model_arch):
 class OVModelForCausalLMIntegrationTest(unittest.TestCase):
     SUPPORTED_ARCHITECTURES = (
         "bart",
-        # "gpt_bigcode",
+        "gpt_bigcode",
         "blenderbot",
         "blenderbot-small",
         "bloom",
