Skip to content

GitLab

Unverified Commit 613c5bb9 authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Bugfix] Fix Aria test loading (#14823) 


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent fd8e055f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
tests/models/decoder_only/vision_language/test_models.py
View file @ 613c5bb9
...@@ -9,7 +9,8 @@ from pathlib import PosixPath ...@@ -9,7 +9,8 @@ from pathlib import PosixPath
import pytest import pytest
from packaging.version import Version from packaging.version import Version
from transformers import AutoModelForPreTraining, AutoModelForVision2Seq from transformers import (AutoModelForImageTextToText, AutoModelForPreTraining,
AutoModelForVision2Seq)
from transformers import __version__ as TRANSFORMERS_VERSION from transformers import __version__ as TRANSFORMERS_VERSION
from vllm.platforms import current_platform from vllm.platforms import current_platform
...@@ -163,6 +164,7 @@ VLM_TEST_SETTINGS = { ...@@ -163,6 +164,7 @@ VLM_TEST_SETTINGS = {
img_idx_to_prompt=lambda idx: "<fim_prefix><|img|><fim_suffix>\n", img_idx_to_prompt=lambda idx: "<fim_prefix><|img|><fim_suffix>\n",
max_model_len=4096, max_model_len=4096,
max_num_seqs=2, max_num_seqs=2,
auto_cls=AutoModelForImageTextToText,
single_image_prompts=IMAGE_ASSETS.prompts({ single_image_prompts=IMAGE_ASSETS.prompts({
"stop_sign": "<vlm_image>Please describe the image shortly.", "stop_sign": "<vlm_image>Please describe the image shortly.",
"cherry_blossom": "<vlm_image>Please infer the season with reason.", "cherry_blossom": "<vlm_image>Please infer the season with reason.",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment