[CI/Build] Fix VLM broadcast tests tensor_parallel_size passing (#10161)

Signed-off-by: Isotr0py <2037008807@qq.com>
2024-11-09 12:02:59 +08:00
parent f83feccd7f
commit 47672f38b5
2 changed files with 2 additions and 0 deletions
--- a/tests/models/decoder_only/vision_language/vlm_utils/types.py
+++ b/tests/models/decoder_only/vision_language/vlm_utils/types.py
@@ -158,6 +158,7 @@ class VLMTestInfo(NamedTuple):
            "max_model_len": self.max_model_len,
            "max_num_seqs": self.max_num_seqs,
            "task": self.task,
+            "tensor_parallel_size": self.tensor_parallel_size,
            "hf_output_post_proc": self.hf_output_post_proc,
            "vllm_output_post_proc": self.vllm_output_post_proc,
            "auto_cls": self.auto_cls,