Skip to content

Commit e59ba9e

Browse files
authored
[CI/Build] Enlarge tolerance for a CPU multi-modal test (#20684)
Signed-off-by: jiang1.li <jiang1.li@intel.com>
1 parent 403b481 commit e59ba9e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/models/multimodal/generation/test_common.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,7 @@
152152
video_idx_to_prompt=lambda idx: "<|vision_bos|><|VIDEO|><|vision_eos|>", # noqa: E501
153153
max_model_len=4096,
154154
max_num_seqs=2,
155+
num_logprobs= 6 if current_platform.is_cpu() else 5,
155156
auto_cls=AutoModelForTextToWaveform,
156157
vllm_output_post_proc=model_utils.qwen2_vllm_to_hf_output,
157158
patch_hf_runner=model_utils.qwen2_5_omni_patch_hf_runner,

0 commit comments

Comments
 (0)