From b1d475f6d249d37cb3d1bf417eaf2b1f8cff2b34 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 4 Sep 2023 11:35:39 +0200 Subject: [PATCH] Skip offload tests for `ViTDet` (#25913) * update * fix --------- Co-authored-by: ydshieh --- tests/models/vitdet/test_modeling_vitdet.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/models/vitdet/test_modeling_vitdet.py b/tests/models/vitdet/test_modeling_vitdet.py index f98c5b4d6..d6ffd03cb 100644 --- a/tests/models/vitdet/test_modeling_vitdet.py +++ b/tests/models/vitdet/test_modeling_vitdet.py @@ -175,6 +175,21 @@ class VitDetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): self.model_tester = VitDetModelTester(self) self.config_tester = ConfigTester(self, config_class=VitDetConfig, has_text_modality=False, hidden_size=37) + # TODO: Fix me (once this model gets more usage) + @unittest.skip("Does not work on the tiny model as we keep hitting edge cases.") + def test_cpu_offload(self): + super().test_cpu_offload() + + # TODO: Fix me (once this model gets more usage) + @unittest.skip("Does not work on the tiny model as we keep hitting edge cases.") + def test_disk_offload(self): + super().test_disk_offload() + + # TODO: Fix me (once this model gets more usage) + @unittest.skip("Does not work on the tiny model as we keep hitting edge cases.") + def test_model_parallelism(self): + super().test_model_parallelism() + def test_config(self): self.config_tester.run_common_tests()