[Bugfix] Remove the last EOS token unless explicitly specified (vllm-…

…project#5077)
blinkbear · Jun 6, 2024 · d58f01f · d58f01f
1 parent b18e919
commit d58f01f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/engine/output_processor/test_stop_checker.py b/tests/engine/output_processor/test_stop_checker.py
@@ -15,7 +15,8 @@ def sequence_with_eos(text: str, eos_token: str,
     """
     seq = Sequence(
         seq_id=0,
-        inputs={"prompt_token_ids": []},
+        prompt="",
+        prompt_token_ids=[],
         block_size=16,
         eos_token_id=eos_token_id,
     )