Skip to content

Commit 75a99b9

Browse files
authored
[Chore] Remove outdated transformers check (#20989)
Signed-off-by: Brayden Zhong <b8zhong@uwaterloo.ca>
1 parent b5c3b68 commit 75a99b9

File tree

1 file changed

+4
-11
lines changed

1 file changed

+4
-11
lines changed

vllm/model_executor/models/idefics3.py

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@
2222

2323
import torch
2424
from torch import nn
25-
from transformers import (AddedToken, BatchFeature, Idefics3Config,
26-
Idefics3ImageProcessor, Idefics3Processor)
25+
from transformers import (BatchFeature, Idefics3Config, Idefics3ImageProcessor,
26+
Idefics3Processor)
2727

2828
from vllm.config import VllmConfig
2929
from vllm.model_executor.layers.linear import ReplicatedLinear
@@ -199,21 +199,14 @@ def get_num_patches(
199199

200200
return grid_w * grid_h + 1
201201

202-
# TODO: Remove after requiring transformers>=4.52
203-
def _get_content(self, token: Union[AddedToken, str]) -> str:
204-
if isinstance(token, str):
205-
return token
206-
207-
return token.content
208-
209202
def _get_image_token(
210203
self,
211204
processor: Optional[Idefics3Processor]) -> tuple[str, str, str]:
212205
if processor is None:
213206
processor = self.get_hf_processor()
214207

215-
image_token = self._get_content(processor.image_token)
216-
fake_image_token = self._get_content(processor.fake_image_token)
208+
image_token = processor.image_token
209+
fake_image_token = processor.fake_image_token
217210
global_image_token = processor.global_image_tag
218211
return image_token, fake_image_token, global_image_token
219212

0 commit comments

Comments
 (0)