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

Merge branch 'master' into release/1.20

Browse Source
This commit is contained in:
Jintao Huang
2024-11-07 16:11:18 +08:00
parent 5c392ba466 19fc9dc3f1
commit 869bf72629
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -53,6 +53,12 @@ def no_multi_modal():
template_info = [
# llama
## "llama3"
TemplateInfo(
template_regex=
f'.*{cases("llama3.2", "llama-3.2")}.*{cases("vision")}.*',
modelfile_prefix=
'https://modelscope.oss-cn-beijing.aliyuncs.com/llm_template/ollama/llama3.2-vision',
),
TemplateInfo(
template=TemplateType.llama3,
template_regex=

3
tests/tools/test_to_ollama.py
View File

@@ -121,6 +121,9 @@ class TestToOllama(unittest.TestCase):
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_check_template_type(self):
_test_check_tmpl_type(
'AI-ModelScope/Llama-3.2-11B-Vision-Instruct-GGUF',
'llama3.2-vision')
_test_check_tmpl_type('LLM-Research/Meta-Llama-3.2-8B-Instruct-GGUF',
'llama3.2')
_test_check_tmpl_type('LLM-Research/Meta-Llama-3.1-8B-Instruct-GGUF',