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

2024-05-29 15:16:41 +08:00 · 2024-05-29 15:16:41 +08:00 · 18c1f16d86
commit 18c1f16d86
parent 5bd3c65072
1 changed files with 5 additions and 2 deletions
--- a/tests/engine/output_processor/test_stop_checker.py
+++ b/tests/engine/output_processor/test_stop_checker.py
@ -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,
    )