Skip to content

Commit fe442e4

Browse files
authored
check (#10587)
1 parent 47de2c6 commit fe442e4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

llm/predict/predictor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1567,7 +1567,7 @@ def send_task_to_queue(task_id):
15671567
else:
15681568
if flag_current_rank_run:
15691569
output_tokens = self.model_inputs["all_token_ids"].numpy()
1570-
output_tokens[output_tokens < 0] = self.tokenizer.pad_token_id
1570+
output_tokens[output_tokens == -1] = self.tokenizer.eos_token_id
15711571
if detokenize:
15721572
outputs = self.tokenizer.batch_decode(
15731573
output_tokens, skip_special_tokens=True, clean_up_tokenization_spaces=False

0 commit comments

Comments
 (0)