From dad2d6a590207cb8938fb915602794665b8e9326 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lucas=20H=C3=A4nke=20de=20Cansino?= Date: Tue, 3 Feb 2026 09:35:58 +0100 Subject: [PATCH] [Bugfix][Model] Fix DeepSeek-OCR-2 chat template to include BOS token (#33642) Signed-off-by: l4b4r4b4b4 --- vllm/transformers_utils/configs/deepseek_vl2.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/vllm/transformers_utils/configs/deepseek_vl2.py b/vllm/transformers_utils/configs/deepseek_vl2.py index 8b02a4ddd..05067c04c 100644 --- a/vllm/transformers_utils/configs/deepseek_vl2.py +++ b/vllm/transformers_utils/configs/deepseek_vl2.py @@ -119,8 +119,9 @@ class DeepseekVLV2Config(PretrainedConfig): self.candidate_resolutions = candidate_resolutions self.vocab_size = self.text_config.vocab_size - # update model_type for OCR model - if "DeepseekOCRForCausalLM" in ( - self.architectures or kwargs.get("architectures", []) - ): + # update model_type for OCR models + architectures = self.architectures or kwargs.get("architectures", []) + if "DeepseekOCRForCausalLM" in architectures: self.model_type = "deepseek_ocr" + elif "DeepseekOCR2ForCausalLM" in architectures: + self.model_type = "deepseek_ocr2"