File tree Expand file tree Collapse file tree 1 file changed +1
-8
lines changed Expand file tree Collapse file tree 1 file changed +1
-8
lines changed Original file line number Diff line number Diff 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 )):
You can’t perform that action at this time.
0 commit comments