Skip to content

GitLab

Unverified Commit 6652e7da authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

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

skip accelerate test
parent dd3a0580
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
tests/models/blip_2/test_modeling_blip_2.py
View file @ 6652e7da
...@@ -710,6 +710,10 @@ class Blip2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): ...@@ -710,6 +710,10 @@ class Blip2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def test_save_load_fast_init_to_base(self): def test_save_load_fast_init_to_base(self):
pass 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): def test_forward_signature(self):
config, _ = self.model_tester.prepare_config_and_inputs_for_common() config, _ = self.model_tester.prepare_config_and_inputs_for_common()
......
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