Skip to content

GitLab

Unverified Commit f27fdfc3 authored by Isotr0py's avatar Isotr0py Committed by GitHub
Browse files

[Bugfix] Investigate Qwen2-VL failing test (#21527) 


Signed-off-by: default avatarIsotr0py <mozf@mail2.sysu.edu.cn>
parent de10ff0b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/models/multimodal/generation/test_common.py
View file @ f27fdfc3
...@@ -677,6 +677,7 @@ VLM_TEST_SETTINGS = { ...@@ -677,6 +677,7 @@ VLM_TEST_SETTINGS = {
prompt_formatter=lambda img_prompt: f"<|im_start|>User\n{img_prompt}<|im_end|>\n<|im_start|>assistant\n", # noqa: E501 prompt_formatter=lambda img_prompt: f"<|im_start|>User\n{img_prompt}<|im_end|>\n<|im_start|>assistant\n", # noqa: E501
img_idx_to_prompt=lambda idx: "<|vision_start|><|image_pad|><|vision_end|>", # noqa: E501 img_idx_to_prompt=lambda idx: "<|vision_start|><|image_pad|><|vision_end|>", # noqa: E501
video_idx_to_prompt=lambda idx: "<|vision_start|><|video_pad|><|vision_end|>", # noqa: E501 video_idx_to_prompt=lambda idx: "<|vision_start|><|video_pad|><|vision_end|>", # noqa: E501
multi_image_prompt="Picture 1: <vlm_image>\nPicture 2: <vlm_image>\nDescribe these two images with one paragraph respectively.", # noqa: E501
max_model_len=4096, max_model_len=4096,
max_num_seqs=2, max_num_seqs=2,
auto_cls=AutoModelForVision2Seq, auto_cls=AutoModelForVision2Seq,
......
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