Skip UnivNetModelTest::test_multi_gpu_data_parallel_forward (#27912)

fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
huggingface · Dec 11, 2023 · e96c1de · e96c1de
1 parent 8d8970e
commit e96c1de
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/models/univnet/test_modeling_univnet.py b/tests/models/univnet/test_modeling_univnet.py
@@ -123,6 +123,10 @@ def setUp(self):
         self.model_tester = UnivNetModelTester(self)
         self.config_tester = ConfigTester(self, config_class=UnivNetConfig)
 
+    @unittest.skip(reason="fix this once it gets more usage")
+    def test_multi_gpu_data_parallel_forward(self):
+        super().test_multi_gpu_data_parallel_forward()
+
     def test_config(self):
         self.config_tester.create_and_test_config_to_json_string()
         self.config_tester.create_and_test_config_to_json_file()