fix: the hardcode hf repo name comparison for deepseek-ocr (#12031)

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Rain Jiang
2025-10-23 21:37:56 -07:00
committed by GitHub
parent aa3003f116
commit 8e797a47f0

View File

@@ -197,10 +197,14 @@ def get_config(
config = AutoConfig.from_pretrained(
model, trust_remote_code=trust_remote_code, revision=revision, **kwargs
)
if "deepseek-ai/DeepSeek-OCR" in model:
if (
getattr(config, "auto_map", None) is not None
and config.auto_map.get("AutoModel")
== "modeling_deepseekocr.DeepseekOCRForCausalLM"
):
config.model_type = "deepseek-ocr"
# Due to an unknown reason, Hugging Faces AutoConfig mistakenly recognizes the configuration of deepseek-ocr as deepseekvl2.
# This is a temporary workaround and will require further optimization.
# TODO: Remove this workaround when AutoConfig correctly identifies deepseek-ocr.
# Hugging Face's AutoConfig currently misidentifies it as deepseekvl2.
except ValueError as e:
if not "deepseek_v32" in str(e):