From ba51aea65e0d2a7afca3f25caba01500fa84d655 Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Wed, 2 Jul 2025 14:46:59 +0800 Subject: [PATCH] [Bugfix] Keye-VL compatibility with `tok_kwargs` (#20058) (#20353) Signed-off-by: DarkLight1337 --- vllm/model_executor/models/keye.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vllm/model_executor/models/keye.py b/vllm/model_executor/models/keye.py index 84996c44aed4..07dcc3a72ad3 100644 --- a/vllm/model_executor/models/keye.py +++ b/vllm/model_executor/models/keye.py @@ -1268,11 +1268,13 @@ class KeyeMultiModalProcessor(BaseMultiModalProcessor[KeyeProcessingInfo]): prompt: str, mm_data: Mapping[str, object], mm_kwargs: Mapping[str, object], + tok_kwargs: Mapping[str, object], ) -> BatchFeature: + mm_kwargs = self.info._get_image_processor_kwargs(**mm_kwargs) return self.info.ctx.call_hf_processor( self.info.get_hf_processor(**mm_kwargs), dict(text=prompt, **mm_data), - self.info._get_image_processor_kwargs(**mm_kwargs), + dict(**mm_kwargs, **tok_kwargs), ) def _get_prompt_updates(