Skip to content

Commit a4a97f1

Browse files
committed
Fix eos
1 parent 2c83f44 commit a4a97f1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integration/resources/test_completion_stream.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def test_create(
3636
) -> None:
3737
prompt = "The space robots have"
3838
model = "meta-llama/Llama-4-Scout-17B-16E-Instruct"
39-
stop = ["</s>"]
39+
stop = ["<|eot|>"]
4040

4141
# max_tokens should be a reasonable number for this test
4242
assert 0 < random_max_tokens < 1024

0 commit comments

Comments
 (0)