Unverified Commit 1887acca authored by Harry Mellor's avatar Harry Mellor Committed by GitHub
Browse files

Fix tokenizer test for renamed attr on Transformers v5 (#33902)


Signed-off-by: default avatarHarry Mellor <19981378+hmellor@users.noreply.github.com>
parent 92e7562a
...@@ -7,6 +7,7 @@ import pytest_asyncio ...@@ -7,6 +7,7 @@ import pytest_asyncio
from transformers import AutoTokenizer from transformers import AutoTokenizer
from vllm.config import ModelConfig from vllm.config import ModelConfig
from vllm.config.utils import getattr_iter
from vllm.v1.engine.detokenizer import check_stop_strings from vllm.v1.engine.detokenizer import check_stop_strings
from ...utils import RemoteOpenAIServer from ...utils import RemoteOpenAIServer
...@@ -131,7 +132,14 @@ async def test_same_response_as_chat_completions(client, tokenizer, messages): ...@@ -131,7 +132,14 @@ async def test_same_response_as_chat_completions(client, tokenizer, messages):
# Post-EOS generation is undefined and may differ # Post-EOS generation is undefined and may differ
eos_tokens = { eos_tokens = {
tokenizer.eos_token_id, tokenizer.eos_token_id,
*tokenizer.additional_special_tokens_ids, *getattr_iter(
tokenizer,
[
"extra_special_tokens_ids", # Transformers v5
"additional_special_tokens_ids", # Transformers v4
],
[],
),
} }
# Find first EOS in generated tokens # Find first EOS in generated tokens
eos_pos = None eos_pos = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment