Skip to content

Commit d93b212

Browse files
committed
fix append
1 parent cfd8e85 commit d93b212

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

fastdeploy/input/ernie4_5_vl_processor/process.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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)

tests/input/test_ernie_vl_processor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)