Skip to content

Commit

Permalink
[Bugfix] Fix arguments passed to Sequence in stop checker test (vll…
Browse files Browse the repository at this point in the history
  • Loading branch information
DarkLight1337 authored May 29, 2024
1 parent 41b73e5 commit d45d516
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions tests/engine/output_processor/test_stop_checker.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,11 @@ def sequence_with_eos(text: str, eos_token: str,
"""
seq = Sequence(
seq_id=0,
prompt="",
prompt_token_ids=[],
inputs={
"prompt": "",
"prompt_token_ids": [],
"multi_modal_data": None,
},
block_size=16,
eos_token_id=eos_token_id,
)
Expand Down

0 comments on commit d45d516

Please sign in to comment.