From abb4308217fc431be18cd114a2b07f0281a356a6 Mon Sep 17 00:00:00 2001
From: suluyana <suluyan_sly@163.com>
Date: Wed, 8 Jan 2025 20:57:53 +0800
Subject: [PATCH] granite

---
 modelscope/preprocessors/templates/loader.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/modelscope/preprocessors/templates/loader.py b/modelscope/preprocessors/templates/loader.py
index d485b352..66029de4 100644
--- a/modelscope/preprocessors/templates/loader.py
+++ b/modelscope/preprocessors/templates/loader.py
@@ -660,6 +660,14 @@ template_info = [
         template_regex=f'.*{cases("granite-embedding")}.*', 
         modelfile_prefix=
         'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/granite-embedding'),
+    TemplateInfo(
+        template_regex=f'.*{cases("granite-3")}.*{cases("2b", "8b")}.*', 
+        modelfile_prefix=
+        'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/granite3-dense'),
+    TemplateInfo(
+        template_regex=f'.*{cases("granite-3")}.*{cases("1b", "3b")}.*', 
+        modelfile_prefix=
+        'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/granite3-moe'),
 
     # opencoder
     TemplateInfo(