Skip to content

Commit

Permalink
Update tests/openvino/test_modeling.py
Browse files Browse the repository at this point in the history
  • Loading branch information
eaidova authored Nov 7, 2024
1 parent a3c6506 commit 8fd1ac5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/openvino/test_modeling.py
Original file line number Diff line number Diff line change
Expand Up @@ -1988,10 +1988,10 @@ def test_compare_to_transformers(self, model_arch):

gc.collect()

@parameterized.expand(["llava", "llava_next"])
@unittest.skipIf(
is_transformers_version("<", "4.45.0"), reason="New preprocessing available only in transformers >= 4.45"
)
@parameterized.expand(["llava", "llava_next"])
def test_llava_with_new_preprocessing(self, model_arch):
prompt = "<image>\n What is shown in this image?"
model_id = MODEL_NAMES[model_arch]
Expand Down

0 comments on commit 8fd1ac5

Please sign in to comment.