Skip to content

Commit 2b0e599

Browse files
committed
use transformers code version
1 parent fffebdb commit 2b0e599

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/openvino/test_modeling.py

+4
Original file line numberDiff line numberDiff line change
@@ -2160,6 +2160,10 @@ def get_transformer_model_class(self, model_arch):
21602160
from transformers import Qwen2VLForConditionalGeneration
21612161

21622162
return Qwen2VLForConditionalGeneration
2163+
if model_arch == "qwen_25_vl":
2164+
from transformers import Qwen2_5_VLForConditionalGeneration
2165+
2166+
return Qwen2_5_VLForConditionalGeneration
21632167
return AutoModelForCausalLM
21642168

21652169
def _check_device_and_request(self, ov_model, expected_device, has_request):

0 commit comments

Comments
 (0)