Skip to content

Commit 36ec5a8

Browse files
DarkLight1337Alvant
authored andcommitted
[Misc] Remove vLLM patch of BaichuanTokenizer (vllm-project#8921)
Signed-off-by: Alvant <alvasian@yandex.ru>
1 parent 6ed2940 commit 36ec5a8

File tree

3 files changed

+3
-273
lines changed

3 files changed

+3
-273
lines changed

vllm/transformers_utils/tokenizer.py

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,7 @@
1111
from vllm.envs import VLLM_USE_MODELSCOPE
1212
from vllm.logger import init_logger
1313
from vllm.lora.request import LoRARequest
14-
from vllm.transformers_utils.tokenizers import (BaichuanTokenizer,
15-
MistralTokenizer)
14+
from vllm.transformers_utils.tokenizers import MistralTokenizer
1615
from vllm.transformers_utils.utils import check_gguf_file
1716
from vllm.utils import make_async
1817

@@ -139,19 +138,6 @@ def get_tokenizer(
139138
raise RuntimeError(err_msg) from e
140139
else:
141140
raise e
142-
except AttributeError as e:
143-
if "BaichuanTokenizer" in str(e):
144-
# This is for the error "'BaichuanTokenizer' object has no
145-
# attribute 'sp_model'".
146-
tokenizer = BaichuanTokenizer.from_pretrained(
147-
tokenizer_name,
148-
*args,
149-
trust_remote_code=trust_remote_code,
150-
revision=revision,
151-
**kwargs,
152-
)
153-
else:
154-
raise e
155141

156142
# NOTE: We can remove this after https://github.com/THUDM/ChatGLM3/issues/1324
157143
if type(tokenizer).__name__ in ("ChatGLMTokenizer",
Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
from vllm.transformers_utils.tokenizers.baichuan import BaichuanTokenizer
2-
from vllm.transformers_utils.tokenizers.mistral import MistralTokenizer
1+
from .mistral import MistralTokenizer
32

4-
__all__ = ["BaichuanTokenizer", "MistralTokenizer"]
3+
__all__ = ["MistralTokenizer"]

vllm/transformers_utils/tokenizers/baichuan.py

Lines changed: 0 additions & 255 deletions
This file was deleted.

0 commit comments

Comments
 (0)