Skip to content

Commit 7af7cdc

Browse files
committed
add test case
1 parent 973f155 commit 7af7cdc

File tree

3 files changed

+10
-1
lines changed

3 files changed

+10
-1
lines changed

optimum/intel/openvino/modeling_visual_language.py

+4
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,10 @@ def prepare_inputs(
165165
if past_len:
166166
position_ids = position_ids[:, -inputs_embeds.shape[1] :]
167167

168+
if self.config.model_type == "qwen2_vl" and position_ids.ndim != 3:
169+
position_ids = np.expand_dims(position_ids, 0)
170+
position_ids = np.concatenate([position_ids, position_ids, position_ids], axis=0)
171+
168172
inputs["position_ids"] = position_ids
169173

170174
if "beam_idx" in self.input_names:

tests/openvino/test_modeling.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -1976,7 +1976,7 @@ class OVModelForVisualCausalLMIntegrationTest(unittest.TestCase):
19761976
if is_transformers_version(">=", "4.40.0"):
19771977
SUPPORTED_ARCHITECTURES += ["llava_next", "nanollava"]
19781978
if is_transformers_version(">=", "4.45.0"):
1979-
SUPPORTED_ARCHITECTURES += ["minicpmv", "internvl2", "phi3_v"]
1979+
SUPPORTED_ARCHITECTURES += ["minicpmv", "internvl2", "phi3_v", "qwen2_vl"]
19801980
TASK = "image-text-to-text"
19811981
REMOTE_CODE_MODELS = ["internvl2", "minicpmv", "nanollava", "phi3_v"]
19821982

@@ -1996,6 +1996,10 @@ def get_transformer_model_class(self, model_arch):
19961996
from transformers import LlavaNextForConditionalGeneration
19971997

19981998
return LlavaNextForConditionalGeneration
1999+
if model_arch == "qwen2_vl":
2000+
from transformers import Qwen2VLForConditionalGeneration
2001+
2002+
return Qwen2VLForConditionalGeneration
19992003
return AutoModelForCausalLM
20002004

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

tests/openvino/utils_tests.py

+1
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,7 @@
115115
"qwen": "katuni4ka/tiny-random-qwen",
116116
"qwen2": "fxmarty/tiny-dummy-qwen2",
117117
"qwen2-moe": "katuni4ka/tiny-random-qwen1.5-moe",
118+
"qwen2_vl": "katuni4ka/tiny-random-qwen2vl",
118119
"resnet": "hf-internal-testing/tiny-random-resnet",
119120
"roberta": "hf-internal-testing/tiny-random-roberta",
120121
"roformer": "hf-internal-testing/tiny-random-roformer",

0 commit comments

Comments
 (0)