[`Blip2`] skip accelerate test (#22124)

skip accelerate test

[`Blip2`] skip accelerate test (#22124)
skip accelerate test
6652e7da · Younes Belkada · GitHub · dd3a0580 · 6652e7da
Unverified Commit 6652e7da authored Mar 13, 2023 by Younes Belkada Committed by GitHub Mar 13, 2023
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

tests/models/blip_2/test_modeling_blip_2.py tests/models/blip_2/test_modeling_blip_2.py +4 -0

No files found.
--- a/tests/models/blip_2/test_modeling_blip_2.py
+++ b/tests/models/blip_2/test_modeling_blip_2.py
@@ -710,6 +710,10 @@ class Blip2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
    def test_save_load_fast_init_to_base(self):
        pass

+    @unittest.skip(reason="Does not work on the tiny model as we keep hitting edge cases.")
+    def test_cpu_offload(self):
+        pass
+
    def test_forward_signature(self):
        config, _ = self.model_tester.prepare_config_and_inputs_for_common()