diff --git a/modelscope/preprocessors/templates/loader.py b/modelscope/preprocessors/templates/loader.py index 97145199..51112432 100644 --- a/modelscope/preprocessors/templates/loader.py +++ b/modelscope/preprocessors/templates/loader.py @@ -771,6 +771,10 @@ template_info = [ template_regex=f'.*{cases("paraphrase-multilingual")}.*', modelfile_prefix= 'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/paraphrase-multilingual'), + TemplateInfo( + template_regex=f'.*{cases("marco")}.*{cases("o1")}.*', + modelfile_prefix= + 'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/marco-o1'), ] diff --git a/tests/tools/test_to_ollama.py b/tests/tools/test_to_ollama.py index 6dbb5f75..b0849e2f 100644 --- a/tests/tools/test_to_ollama.py +++ b/tests/tools/test_to_ollama.py @@ -298,6 +298,7 @@ class TestToOllama(unittest.TestCase): _test_check_tmpl_type( 'Ceceliachenen/paraphrase-multilingual-MiniLM-L12-v2', 'paraphrase-multilingual') + _test_check_tmpl_type('bartowski/Marco-o1-GGUF', 'marco-o1') if __name__ == '__main__':