From ecfcdd2ce47e2216eee11550645b1bb3cfa44d7b Mon Sep 17 00:00:00 2001 From: Harry Mellor <19981378+hmellor@users.noreply.github.com> Date: Tue, 17 Mar 2026 14:29:24 +0000 Subject: [PATCH] Fix Phi3 test that fails with Transformers v5 (#37298) Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> --- tests/models/multimodal/pooling/test_phi3v.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/models/multimodal/pooling/test_phi3v.py b/tests/models/multimodal/pooling/test_phi3v.py index c799a5bd3..2794b0b29 100644 --- a/tests/models/multimodal/pooling/test_phi3v.py +++ b/tests/models/multimodal/pooling/test_phi3v.py @@ -3,6 +3,7 @@ import pytest import torch.nn.functional as F +import transformers.utils from PIL import Image from vllm.assets.base import get_vllm_public_assets @@ -12,6 +13,12 @@ from ....conftest import IMAGE_ASSETS, HfRunner, PromptImageInput, VllmRunner from ....utils import large_gpu_test from ...utils import check_embeddings_close +# BC for method that was deleted in Transformers v5. +# Only needed for generating the HF reference. +transformers.utils.is_flash_attn_greater_or_equal_2_10 = ( + lambda: transformers.utils.is_flash_attn_greater_or_equal("2.1.0") +) + HF_TEXT_PROMPTS = [ # T -> X "Find me an everyday image that matches the given caption: The label of the object is stop sign", # noqa: E501