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

fix ci

Browse Source
This commit is contained in:
suluyana
2024-11-04 10:41:21 +08:00
parent 7a75751c17
commit 3c5efc1669
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -198,7 +198,7 @@ template_info = [
TemplateInfo(
template=TemplateType.gemma,
template_regex=
f'.*{cases("gemma")}.*',
f'{no("pali")}.*{cases("gemma")}.*',
modelfile_prefix=
'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/gemma',
),

2
tests/tools/test_to_ollama.py
View File

@@ -63,7 +63,7 @@ class TestToOllama(unittest.TestCase):
template = TemplateLoader.load_by_model_id(
'AI-ModelScope/gemma-2b-instruct')
self.assertTrue(template is None)
self.assertTrue(template.template_type == TemplateType.gemma)
template = TemplateLoader.load_by_model_id(
'AI-ModelScope/gemma2-2b-instruct')