Unverified Commit 8f76dc8e authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Skip failing `AlignModelTest::test_multi_gpu_data_parallel_forward` (#23374)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 41d47db9
...@@ -437,6 +437,10 @@ class AlignModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): ...@@ -437,6 +437,10 @@ class AlignModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
config_and_inputs = self.model_tester.prepare_config_and_inputs() config_and_inputs = self.model_tester.prepare_config_and_inputs()
self.model_tester.create_and_check_model(*config_and_inputs) self.model_tester.create_and_check_model(*config_and_inputs)
@unittest.skip(reason="Start to fail after using torch `cu118`.")
def test_multi_gpu_data_parallel_forward(self):
super().test_multi_gpu_data_parallel_forward()
@unittest.skip(reason="Hidden_states is tested in individual model tests") @unittest.skip(reason="Hidden_states is tested in individual model tests")
def test_hidden_states_output(self): def test_hidden_states_output(self):
pass pass
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment