From 23f3b9a43eae45e76cf012dba3afb46d1a5e20de Mon Sep 17 00:00:00 2001
From: "xiaojianping.xjp" <1015212746@QQ.COM>
Date: Fri, 7 Nov 2025 17:33:25 +0800
Subject: [PATCH] update ollama: granite4,deepseek-v3.1,gpt-oss (#1531)

Co-authored-by: xiaojiaping.xjp <xiaojiaping.xjp@alibaba-inc.com>
---
 modelscope/preprocessors/templates/loader.py | 12 ++++++++++++
 tests/tools/test_to_ollama.py                | 12 ++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/modelscope/preprocessors/templates/loader.py b/modelscope/preprocessors/templates/loader.py
index f9c8f944..d4dca7bb 100644
--- a/modelscope/preprocessors/templates/loader.py
+++ b/modelscope/preprocessors/templates/loader.py
@@ -194,6 +194,10 @@ template_info = [
         modelfile_prefix=
         'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/deepseek-llm',
     ),
+    TemplateInfo(
+        template_regex=f'.*{cases("DeepSeek-V3.1")}.*',
+        modelfile_prefix=
+        'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/deepseek-v3.1'),
     TemplateInfo(
         template_regex=
         f'.*{cases("deepseek")}.*{cases("v3")}.*',
@@ -994,6 +998,14 @@ template_info = [
         template_regex=f'.*{cases("deepscaler")}.*',
         modelfile_prefix=
         'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/deepscaler'),
+    TemplateInfo(
+        template_regex=f'.*{cases("granite-4.0")}.*',
+        modelfile_prefix=
+        'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/granite4'),
+    TemplateInfo(
+        template_regex=f'.*{cases("gpt-oss")}.*',
+        modelfile_prefix=
+        'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/gpt-oss'),
 ]
 
 
diff --git a/tests/tools/test_to_ollama.py b/tests/tools/test_to_ollama.py
index 274fd56d..e3e925f0 100644
--- a/tests/tools/test_to_ollama.py
+++ b/tests/tools/test_to_ollama.py
@@ -122,6 +122,18 @@ class TestToOllama(unittest.TestCase):
 
     @unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
     def test_check_template_type(self):
+        _test_check_tmpl_type(
+            'unsloth/gpt-oss-20b-GGUF',
+            'gpt-oss',
+            gguf_meta={'general.name': 'Gpt-Oss-20B'})
+        _test_check_tmpl_type(
+            'unsloth/granite-4.0-h-tiny-GGUF',
+            'granite4',
+            gguf_meta={'general.name': 'Granite-4.0-H-Tiny'})
+        _test_check_tmpl_type(
+            'unsloth/DeepSeek-V3.1-GGUF',
+            'deepseek-v3.1',
+            gguf_meta={'general.name': 'Deepseek-V3.1'})
         _test_check_tmpl_type('unsloth/Qwen3-Coder-30B-A3B-Instruct-GGUF',
                               'qwen3-coder')
         _test_check_tmpl_type(