Skip to content

Commit 233557b

Browse files
committed
fix
1 parent 1c6c79c commit 233557b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/openvino/test_modeling.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -1119,7 +1119,7 @@ def test_compare_to_transformers(self, model_arch):
11191119
ov_outputs = ov_model.generate(**tokens, generation_config=gen_config)
11201120

11211121
# TODO: add back once https://huggingface.co/katuni4ka/tiny-random-minicpm3/discussions/1 merged (for all models) as current mdoeling incompatible with transformers >= v4.49
1122-
if model_arch in {"minicpm", "minicpm3", "arctic", "deepseek"}:
1122+
if model_arch in {"minicpm", "minicpm3", "arctic", "deepseek"} and is_transformers_version(">=", "4.49"):
11231123
pass
11241124

11251125
additional_inputs = {}
@@ -2227,11 +2227,11 @@ def test_compare_to_transformers(self, model_arch):
22272227
set_seed(SEED)
22282228

22292229
# TODO: add back once https://huggingface.co/katuni4ka/tiny-random-maira2/discussions/1 merged as current mdoeling incompatible with transformers >= v4.49
2230-
if model_arch in {"maira2"}:
2230+
if model_arch in {"maira2"} and is_transformers_version(">=", "4.49"):
22312231
pass
22322232

22332233
# TODO: add back once https://huggingface.co/katuni4ka/tiny-random-minicpm3/discussions/1 merged for all models as current mdoeling incompatible with transformers >= v4.49
2234-
if model_arch in {"phi3_v"}:
2234+
if model_arch in {"phi3_v"} and is_transformers_version(">=", "4.49"):
22352235
pass
22362236

22372237
with torch.no_grad():

0 commit comments

Comments
 (0)