Qwen-2-5-VL: fix CI (#35935)

fix
This commit is contained in:
Raushan Turganbay
2025-01-28 14:51:57 +01:00
committed by GitHub
parent 3f860dba55
commit f85ba20449

View File

@@ -1655,7 +1655,7 @@ class GenerationTesterMixin:
inputs_dict.pop("pixel_values_images", None)
# 2.C - No easy fix, let's skip the check that compares the outputs from `input_ids` and `inputs_embeds`
has_complex_embeds_computation = any(
model_name in model_class.__name__.lower() for model_name in ["moshi", "qwen2vl"]
model_name in model_class.__name__.lower() for model_name in ["moshi", "qwen2vl", "qwen2_5_vl"]
)
# 3 - `inputs_dict` doesn't contain `attention_mask`. When `attention_mask` is not passed to generate,
# we infer it from `input_ids`. The last test case will fail if there is a pad token in the original input.