skip test_model_parallelism for 2 model test classes (huggingface#3…

…1067) skip Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
vasqu · Jun 1, 2024 · 04440a0 · 04440a0
1 parent f803e2b
commit 04440a0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/tests/models/deformable_detr/test_modeling_deformable_detr.py b/tests/models/deformable_detr/test_modeling_deformable_detr.py
@@ -232,6 +232,12 @@ def setUp(self):
         self.model_tester = DeformableDetrModelTester(self)
         self.config_tester = ConfigTester(self, config_class=DeformableDetrConfig, has_text_modality=False)
 
+    @unittest.skip(
+        "This starts to fail since 2024/05/24, but earlier commits also fail now and affect many other tests. The error is `an illegal memory access was encountered`."
+    )
+    def test_model_parallelism(self):
+        super().test_model_parallelism()
+
     def test_config(self):
         # we don't test common_properties and arguments_init as these don't apply for Deformable DETR
         self.config_tester.create_and_test_config_to_json_string()

diff --git a/tests/models/rwkv/test_modeling_rwkv.py b/tests/models/rwkv/test_modeling_rwkv.py
@@ -304,6 +304,12 @@ def assertInterval(self, member, container, msg=None):
             standardMsg = "%s not found in %s" % (safe_repr(member), safe_repr(container))
             self.fail(self._formatMessage(msg, standardMsg))
 
+    @unittest.skip(
+        "This starts to fail since 2024/05/24, but earlier commits also fail now and affect many other tests. The error is `an illegal memory access was encountered`."
+    )
+    def test_model_parallelism(self):
+        super().test_model_parallelism()
+
     def test_config(self):
         self.config_tester.run_common_tests()