fix prompt_token_ids is None in request dict (#5241)

This commit is contained in:
kxz2002
2025-11-26 17:10:45 +08:00
committed by GitHub
parent 00d0ef5134
commit bc118c3d2d

View File

@@ -263,7 +263,9 @@ class Ernie4_5_VLProcessor(Ernie4_5Processor):
outputs = self.pack_outputs(outputs)
request["prompt_token_ids"] = (
outputs["input_ids"].tolist() if "prompt_token_ids" not in request else request["prompt_token_ids"]
outputs["input_ids"].tolist()
if ("prompt_token_ids" not in request or not request["prompt_token_ids"])
else request["prompt_token_ids"]
)
request["prompt_token_ids_len"] = len(request["prompt_token_ids"])
request["multimodal_inputs"] = outputs