Unverified Commit bc2ae5a3 authored by Angela Yi's avatar Angela Yi Committed by GitHub
Browse files

[Test] Increase qwen2_vl num_logprobs to fix torch 2.12 update (#40818)


Signed-off-by: default avatarAngela Yi <angelayi@meta.com>
parent a474da28
......@@ -921,6 +921,7 @@ VLM_TEST_SETTINGS = {
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_num_seqs=2,
num_logprobs=10,
auto_cls=AutoModelForImageTextToText,
vllm_output_post_proc=model_utils.qwen2_vllm_to_hf_output,
image_size_factors=[(0.25,), (0.25, 0.25, 0.25), (0.25, 0.2, 0.15)],
......
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