Skip to content

Commit 0987dfa

Browse files
committed
Done
Signed-off-by: Jee Jee Li <[email protected]>
1 parent 63d2705 commit 0987dfa

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

tests/lora/test_minicpmv_tp.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -66,8 +66,7 @@ def test_minicpmv_lora(minicpmv_lora_files):
6666
max_loras=2,
6767
max_lora_rank=8,
6868
enforce_eager=True,
69-
trust_remote_code=True,
70-
enable_chunked_prefill=True,
69+
max_model_len=2048,
7170
)
7271
output1 = do_sample(llm, minicpmv_lora_files, lora_id=1)
7372
for i in range(len(EXPECTED_OUTPUT)):
@@ -91,9 +90,6 @@ def test_minicpmv_tp4_wo_fully_sharded_loras(minicpmv_lora_files):
9190
max_loras=4,
9291
max_lora_rank=64,
9392
tensor_parallel_size=4,
94-
trust_remote_code=True,
95-
enforce_eager=True,
96-
enable_chunked_prefill=True,
9793
)
9894
output_tp = do_sample(llm, minicpmv_lora_files, lora_id=1)
9995
for i in range(len(EXPECTED_OUTPUT)):
@@ -114,9 +110,6 @@ def test_minicpmv_tp4_fully_sharded_loras(minicpmv_lora_files):
114110
max_loras=2,
115111
max_lora_rank=8,
116112
tensor_parallel_size=4,
117-
trust_remote_code=True,
118-
fully_sharded_loras=True,
119-
enable_chunked_prefill=True,
120113
)
121114
output_tp = do_sample(llm, minicpmv_lora_files, lora_id=1)
122115
for i in range(len(EXPECTED_OUTPUT)):

0 commit comments

Comments
 (0)