From d0c08e36a51d1c8974580cbd8ce4f15c207ff065 Mon Sep 17 00:00:00 2001
From: bluebread <hotbread70127@gmail.com>
Date: Fri, 5 Dec 2025 04:03:56 +0000
Subject: [PATCH] mtmd: minor fix

---
 src/models/deepseek2.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/models/deepseek2.cpp b/src/models/deepseek2.cpp
index 999ee96b2e..c3ea9b7843 100644
--- a/src/models/deepseek2.cpp
+++ b/src/models/deepseek2.cpp
@@ -6,7 +6,7 @@ llm_build_deepseek2::llm_build_deepseek2(const llama_model & model, const llm_gr
     llm_graph_context(params) {
     // lite variants include DeepSeek-V2-Lite, GigaChat3-10B-A1.8B
     bool is_lite = (hparams.n_layer == 27 || hparams.n_layer == 26);
-    bool is_ocr = (model.name.find("ocr") != std::string::npos || model.name.find("OCR") != std::string::npos);
+    bool is_ocr = (model.arch == LLM_ARCH_DEEPSEEK2OCR);
 
     const bool is_mla = (hparams.n_embd_head_k_mla != 0 && hparams.n_embd_head_v_mla != 0);