xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-15 05:15:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[CI] Move testing image from remote URL to S3 (#23980)

Browse Source 
Signed-off-by: Roger Wang <hey@rogerw.io>
This commit is contained in:
Roger Wang 2025-08-29 21:41:25 -07:00 committed by GitHub
parent 8fb85b7bb6
commit ee52a32705
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/models/multimodal/generation/vlm_utils/custom_inputs.py
View File

@ -1,12 +1,9 @@
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Custom input builders for edge-cases in different models.""" """Custom input builders for edge-cases in different models."""
from io import BytesIO
from typing import Callable from typing import Callable
import requests from vllm.assets.image import ImageAsset
from PIL import Image
from vllm.multimodal.image import rescale_image_size from vllm.multimodal.image import rescale_image_size
from vllm.multimodal.video import (rescale_video_size, resize_video, from vllm.multimodal.video import (rescale_video_size, resize_video,
sample_frames_from_video) sample_frames_from_video)
@ -118,9 +115,9 @@ def different_patch_input_cases_internvl():
def windows_attention_image_qwen2_5_vl(): def windows_attention_image_qwen2_5_vl():
# image from regression issue: https://github.com/vllm-project/vllm/issues/15122
image_url = "https://github.com/AOMediaCodec/av1-avif/blob/main/testFiles/Link-U/hato.jpg?raw=true" # image from regression issue: https://github.com/vllm-project/vllm/issues/15122 # noqa: E501
image = Image.open(BytesIO(requests.get(image_url).content)) image = ImageAsset("hato").pil_image
question = "Describe the image." question = "Describe the image."
img_prompt = "<|vision_start|><|image_pad|><|vision_end|>" img_prompt = "<|vision_start|><|image_pad|><|vision_end|>"

2
vllm/assets/image.py
View File

@ -11,7 +11,7 @@ from .base import get_vllm_public_assets
VLM_IMAGES_DIR = "vision_model_images" VLM_IMAGES_DIR = "vision_model_images"
ImageAssetName = Literal["stop_sign", "cherry_blossom"] ImageAssetName = Literal["stop_sign", "cherry_blossom", "hato"]
@dataclass(frozen=True) @dataclass(frozen=True)