fix / skip (for now) some tests before switch to torch 2.2 (#28838)
* fix / skip some tests before we can switch to torch 2.2 * style --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -176,6 +176,11 @@ class VitsModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
|
||||
def test_config(self):
|
||||
self.config_tester.run_common_tests()
|
||||
|
||||
# TODO: @ydshieh
|
||||
@is_flaky(description="torch 2.2.0 gives `Timeout >120.0s`")
|
||||
def test_pipeline_feature_extraction(self):
|
||||
super().test_pipeline_feature_extraction()
|
||||
|
||||
@unittest.skip("Need to fix this after #26538")
|
||||
def test_model_forward(self):
|
||||
set_seed(12345)
|
||||
|
||||
Reference in New Issue
Block a user