File tree Expand file tree Collapse file tree 1 file changed +13
-4
lines changed Expand file tree Collapse file tree 1 file changed +13
-4
lines changed Original file line number Diff line number Diff line change @@ -66,8 +66,12 @@ def test_minicpmv_lora(minicpmv_lora_files):
6666 max_loras = 2 ,
6767 max_lora_rank = 8 ,
6868 enforce_eager = True ,
69+ max_model_len = 2048 ,
70+ limit_mm_per_prompt = {
71+ "image" : 2 ,
72+ "video" : 0
73+ },
6974 trust_remote_code = True ,
70- enable_chunked_prefill = True ,
7175 )
7276 output1 = do_sample (llm , minicpmv_lora_files , lora_id = 1 )
7377 for i in range (len (EXPECTED_OUTPUT )):
@@ -91,9 +95,11 @@ def test_minicpmv_tp4_wo_fully_sharded_loras(minicpmv_lora_files):
9195 max_loras = 4 ,
9296 max_lora_rank = 64 ,
9397 tensor_parallel_size = 4 ,
98+ limit_mm_per_prompt = {
99+ "image" : 2 ,
100+ "video" : 0
101+ },
94102 trust_remote_code = True ,
95- enforce_eager = True ,
96- enable_chunked_prefill = True ,
97103 )
98104 output_tp = do_sample (llm , minicpmv_lora_files , lora_id = 1 )
99105 for i in range (len (EXPECTED_OUTPUT )):
@@ -115,8 +121,11 @@ def test_minicpmv_tp4_fully_sharded_loras(minicpmv_lora_files):
115121 max_lora_rank = 8 ,
116122 tensor_parallel_size = 4 ,
117123 trust_remote_code = True ,
124+ limit_mm_per_prompt = {
125+ "image" : 1 ,
126+ "video" : 0
127+ },
118128 fully_sharded_loras = True ,
119- enable_chunked_prefill = True ,
120129 )
121130 output_tp = do_sample (llm , minicpmv_lora_files , lora_id = 1 )
122131 for i in range (len (EXPECTED_OUTPUT )):
You can’t perform that action at this time.
0 commit comments