File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
fastdeploy/input/ernie4_5_vl_processor Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -356,7 +356,7 @@ def prompt_token_ids2outputs(
356356 prompt_token_ids = request .get ("prompt_token_ids" , [])
357357 prompt_token_ids_len = len (prompt_token_ids )
358358 if not request .get ("messages" ):
359- outputs ["input_ids" ].append (prompt_token_ids )
359+ outputs ["input_ids" ].extend (prompt_token_ids )
360360 outputs ["token_type_ids" ].extend ([IDS_TYPE_FLAG ["text" ]] * prompt_token_ids_len )
361361 for i in range (prompt_token_ids_len ):
362362 outputs ["position_ids" ].append ([i ] * 3 )
Original file line number Diff line number Diff line change @@ -194,7 +194,7 @@ def test_prompt_token_ids2outputs_only_prompt_token_ids(self):
194194
195195 self .assertEqual (
196196 outputs ["input_ids" ],
197- [ test_prompt_token_ids ] ,
197+ test_prompt_token_ids ,
198198 f"input_ids 不匹配:实际{ outputs ['input_ids' ]} ,预期[{ test_prompt_token_ids } ]" ,
199199 )
200200
You can’t perform that action at this time.
0 commit comments