diff --git a/tests/models/qwen2_vl/test_modeling_qwen2_vl.py b/tests/models/qwen2_vl/test_modeling_qwen2_vl.py index 51c2f7c2c61..222fa20d0a2 100644 --- a/tests/models/qwen2_vl/test_modeling_qwen2_vl.py +++ b/tests/models/qwen2_vl/test_modeling_qwen2_vl.py @@ -18,6 +18,7 @@ import unittest import requests +from parameterized import parameterized from transformers import ( AutoProcessor, @@ -337,9 +338,7 @@ def test_generate_compile_model_forward(self): pass @parameterized.expand([("greedy", 1), ("beam search", 2)]) - @unittest.skip( - reason="Breaks the CIs right now! To check!" - ) + @unittest.skip(reason="Breaks the CIs right now! To check!") def test_generate_from_inputs_embeds(self, _, num_beams): pass