Skip to content

Commit d629618

Browse files
jsato8094blinkbear
authored andcommitted
[Bugfix] Remove the last EOS token unless explicitly specified (vllm-project#5077)
1 parent c859d82 commit d629618

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

tests/engine/output_processor/test_stop_checker.py

+2-5
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,8 @@ def sequence_with_eos(text: str, eos_token: str,
1515
"""
1616
seq = Sequence(
1717
seq_id=0,
18-
inputs={
19-
"prompt": "",
20-
"prompt_token_ids": [],
21-
"multi_modal_data": None,
22-
},
18+
prompt="",
19+
prompt_token_ids=[],
2320
block_size=16,
2421
eos_token_id=eos_token_id,
2522
)

0 commit comments

Comments
 (0)