Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

add marco o1

Browse Source
This commit is contained in:
suluyana
2024-11-25 14:21:41 +08:00
parent e3f63fd1ea
commit e7c36637a7
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/preprocessors/templates/loader.py
View File

@@ -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'),
]

1
tests/tools/test_to_ollama.py
View File

@@ -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__':