Skip to content

Commit bc118c3

Browse files
authored
fix prompt_token_ids is None in request dict (PaddlePaddle#5241)
1 parent 00d0ef5 commit bc118c3

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

fastdeploy/input/ernie4_5_vl_processor/ernie4_5_vl_processor.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,7 +263,9 @@ def process_request_dict(self, request, max_model_len=None):
263263

264264
outputs = self.pack_outputs(outputs)
265265
request["prompt_token_ids"] = (
266-
outputs["input_ids"].tolist() if "prompt_token_ids" not in request else request["prompt_token_ids"]
266+
outputs["input_ids"].tolist()
267+
if ("prompt_token_ids" not in request or not request["prompt_token_ids"])
268+
else request["prompt_token_ids"]
267269
)
268270
request["prompt_token_ids_len"] = len(request["prompt_token_ids"])
269271
request["multimodal_inputs"] = outputs

0 commit comments

Comments
 (0)