Unverified Commit e936e401 authored by Isotr0py's avatar Isotr0py Committed by GitHub
Browse files

[Bugfix] Fix processor initialization in transformers 4.53.0 (#20244)


Signed-off-by: default avatarIsotr0py <2037008807@qq.com>
parent f5dfa075
...@@ -5,7 +5,9 @@ from dataclasses import dataclass ...@@ -5,7 +5,9 @@ from dataclasses import dataclass
from typing import TYPE_CHECKING, Any, NamedTuple, Optional, Union from typing import TYPE_CHECKING, Any, NamedTuple, Optional, Union
import torch import torch
from packaging.version import Version
from transformers import BatchFeature, PretrainedConfig, ProcessorMixin from transformers import BatchFeature, PretrainedConfig, ProcessorMixin
from transformers import __version__ as TRANSFORMERS_VERSION
from typing_extensions import TypeVar from typing_extensions import TypeVar
from vllm.jsontree import JSONTree, json_map_leaves from vllm.jsontree import JSONTree, json_map_leaves
...@@ -128,9 +130,13 @@ class InputProcessingContext(InputContext): ...@@ -128,9 +130,13 @@ class InputProcessingContext(InputContext):
/, /,
**kwargs: object, **kwargs: object,
) -> _P: ) -> _P:
# Transformers 4.53.0 has issue with passing tokenizer to
# initialize processor. We disable it for this version.
# See: https://github.com/vllm-project/vllm/issues/20224
if Version(TRANSFORMERS_VERSION) != Version("4.53.0"):
kwargs["tokenizer"] = self.tokenizer
return super().get_hf_processor( return super().get_hf_processor(
typ, typ,
tokenizer=self.tokenizer,
**kwargs, **kwargs,
) )
......
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