From 3c5efc1669826ff55baf6f51368fce6e2cd5f393 Mon Sep 17 00:00:00 2001 From: suluyana Date: Mon, 4 Nov 2024 10:41:21 +0800 Subject: [PATCH] fix ci --- modelscope/preprocessors/templates/loader.py | 2 +- tests/tools/test_to_ollama.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/modelscope/preprocessors/templates/loader.py b/modelscope/preprocessors/templates/loader.py index bc5fc0ee..ccbd6e09 100644 --- a/modelscope/preprocessors/templates/loader.py +++ b/modelscope/preprocessors/templates/loader.py @@ -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', ), diff --git a/tests/tools/test_to_ollama.py b/tests/tools/test_to_ollama.py index e05b7e42..d64af6dd 100644 --- a/tests/tools/test_to_ollama.py +++ b/tests/tools/test_to_ollama.py @@ -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')