Skip to content

Commit 7b37ecc

Browse files
princepridehuydhn
authored andcommitted
Add ignore consolidated file in mistral example code (vllm-project#20420)
Signed-off-by: 汪志鹏 <wangzhipeng628@gmail.com>
1 parent 59a6a96 commit 7b37ecc

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

examples/offline_inference/vision_language.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -677,6 +677,7 @@ def run_mistral3(questions: list[str], modality: str) -> ModelRequestData:
677677
max_num_seqs=2,
678678
tensor_parallel_size=2,
679679
limit_mm_per_prompt={modality: 1},
680+
ignore_patterns=["consolidated.safetensors"],
680681
)
681682

682683
prompts = [f"<s>[INST]{question}\n[IMG][/INST]" for question in questions]

examples/offline_inference/vision_language_multi_image.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -505,6 +505,7 @@ def load_mistral3(question: str, image_urls: list[str]) -> ModelRequestData:
505505
max_num_seqs=2,
506506
tensor_parallel_size=2,
507507
limit_mm_per_prompt={"image": len(image_urls)},
508+
ignore_patterns=["consolidated.safetensors"],
508509
)
509510

510511
placeholders = "[IMG]" * len(image_urls)

0 commit comments

Comments
 (0)