Skip to content

Commit e806a32

Browse files
committed
Fix up vision test
1 parent c4ef4b6 commit e806a32

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

.jenkins/vision/configs/ovis2_5-9b.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,4 @@ max_model_len: 32768
44
max_num_seqs: 32
55
num_prompts: 4
66
limit_mm_per_prompt_image: 5
7+
trust_remote_code: True

.jenkins/vision/test_enc_dec_model.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ def fail_on_exit():
2424
def launch_enc_dec_model(config, question, images):
2525
model_name = config.get('model_name')
2626
dtype = config.get('dtype', 'bfloat16')
27+
trust_remote_code = config.get('trust_remote_code', False)
2728
max_num_seqs = config.get('max_num_seqs', 128)
2829
max_model_len = config.get('max_model_len', 4096)
2930
enforce_eager = config.get('enforce_eager', False)
@@ -41,6 +42,7 @@ def launch_enc_dec_model(config, question, images):
4142
enable_expert_parallel=enable_expert_parallel,
4243
enforce_eager=enforce_eager,
4344
limit_mm_per_prompt={"image": limit_mm_per_prompt_image},
45+
trust_remote_code=trust_remote_code,
4446
)
4547

4648
tokenizer = AutoTokenizer.from_pretrained(model_name)

0 commit comments

Comments
 (0)