"vscode:/vscode.git/clone" did not exist on "8e0ae19da7961a9b83f24d58f9f4e672f71f24db"
Unverified Commit dad2d6a5 authored by Lucas Hänke de Cansino's avatar Lucas Hänke de Cansino Committed by GitHub
Browse files

[Bugfix][Model] Fix DeepSeek-OCR-2 chat template to include BOS token (#33642)


Signed-off-by: default avatarl4b4r4b4b4 <lucas.cansino@mail.de>
parent 32e84fa1
......@@ -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"
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