Gahow Wang
445e491123
third_party/vllm/ now tracked in git for direct patch management.
Based on vLLM v0.18.1 release with one patch applied:
vllm/v1/core/sched/scheduler.py:
Replace fatal assert with graceful skip when KV transfer callback
arrives for an already-aborted request during PD disaggregated serving.
Future vLLM modifications should be made directly in third_party/vllm/
and committed normally. The patches/ directory is kept as documentation
of what changed from upstream.
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
from typing import _get_protocol_attrs # type: ignore
|
|
|
|
import pytest
|
|
from transformers import (
|
|
PreTrainedTokenizerBase,
|
|
PreTrainedTokenizerFast,
|
|
)
|
|
|
|
from vllm.tokenizers import TokenizerLike, get_tokenizer
|
|
from vllm.tokenizers.grok2 import Grok2Tokenizer
|
|
from vllm.tokenizers.hf import HfTokenizer
|
|
from vllm.tokenizers.mistral import MistralTokenizer
|
|
|
|
|
|
def _get_missing_attrs(obj: object, target: type):
|
|
return [k for k in _get_protocol_attrs(target) if not hasattr(obj, k)]
|
|
|
|
|
|
def _assert_tokenizer_like(tokenizer: object):
|
|
missing_attrs = _get_missing_attrs(tokenizer, TokenizerLike)
|
|
assert not missing_attrs, f"Missing attrs: {missing_attrs}"
|
|
|
|
|
|
def test_tokenizer_like_protocol():
|
|
tokenizer = get_tokenizer("gpt2", use_fast=True)
|
|
assert isinstance(tokenizer, PreTrainedTokenizerFast)
|
|
_assert_tokenizer_like(tokenizer)
|
|
|
|
tokenizer = get_tokenizer(
|
|
"mistralai/Mistral-7B-Instruct-v0.3",
|
|
tokenizer_mode="mistral",
|
|
)
|
|
assert isinstance(tokenizer, MistralTokenizer)
|
|
_assert_tokenizer_like(tokenizer)
|
|
|
|
tokenizer = get_tokenizer("xai-org/grok-2", tokenizer_mode="grok2")
|
|
assert isinstance(tokenizer, Grok2Tokenizer)
|
|
_assert_tokenizer_like(tokenizer)
|
|
|
|
tokenizer = get_tokenizer("deepseek-ai/DeepSeek-V3", tokenizer_mode="deepseek_v32")
|
|
assert isinstance(tokenizer, HfTokenizer)
|
|
|
|
# Verify it's a fast tokenizer (required for FastIncrementalDetokenizer)
|
|
assert isinstance(tokenizer, PreTrainedTokenizerFast)
|
|
assert "DSV32" in tokenizer.__class__.__name__
|
|
_assert_tokenizer_like(tokenizer)
|
|
|
|
tokenizer = get_tokenizer(
|
|
"Qwen/Qwen-VL",
|
|
tokenizer_mode="qwen_vl",
|
|
trust_remote_code=True,
|
|
)
|
|
assert isinstance(tokenizer, HfTokenizer)
|
|
assert "WithoutImagePad" in tokenizer.__class__.__name__
|
|
|
|
|
|
@pytest.mark.parametrize("tokenizer_name", ["facebook/opt-125m", "gpt2"])
|
|
def test_tokenizer_revision(tokenizer_name: str):
|
|
# Assume that "main" branch always exists
|
|
tokenizer = get_tokenizer(tokenizer_name, revision="main")
|
|
assert isinstance(tokenizer, PreTrainedTokenizerBase)
|
|
|
|
# Assume that "never" branch always does not exist
|
|
with pytest.raises(OSError, match="not a valid git identifier"):
|
|
get_tokenizer(tokenizer_name, revision="never")
|
|
|
|
|
|
@pytest.mark.parametrize("tokenizer_name", ["BAAI/bge-base-en"])
|
|
@pytest.mark.parametrize("n_tokens", [510])
|
|
def test_special_tokens(tokenizer_name: str, n_tokens: int):
|
|
tokenizer = get_tokenizer(tokenizer_name, revision="main")
|
|
|
|
prompts = "[UNK]" * n_tokens
|
|
prompt_token_ids = tokenizer.encode(prompts)
|
|
assert len(prompt_token_ids) == n_tokens + 2
|