20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Bugfix] Fix arguments passed to Sequence in stop checker test (#5092)

Browse Source
This commit is contained in:
Cyrus Leung 2024-05-29 15:16:41 +08:00 committed by GitHub
parent 5bd3c65072
commit 18c1f16d86
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/engine/output_processor/test_stop_checker.py
View File

@ -15,8 +15,11 @@ def sequence_with_eos(text: str, eos_token: str,
""" """
seq = Sequence( seq = Sequence(
seq_id=0, seq_id=0,
prompt="", inputs={
prompt_token_ids=[], "prompt": "",
"prompt_token_ids": [],
"multi_modal_data": None,
},
block_size=16, block_size=16,
eos_token_id=eos_token_id, eos_token_id=eos_token_id,
) )