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

fix test case

Browse Source
This commit is contained in:
suluyan
2025-01-21 12:36:57 +08:00
parent 6eeb9bff15
commit f0aaca461b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/tools/test_to_ollama.py
View File

@@ -16,7 +16,7 @@ def _test_check_tmpl_type(model, tmpl_type, gguf_meta={}):
class TestToOllama(unittest.TestCase):
@unittest.skip#Unless(test_level() >= 0, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_load_template(self):
template = TemplateLoader.load_by_model_id(
'LLM-Research/Meta-Llama-3-8B-Instruct')
@@ -86,7 +86,7 @@ class TestToOllama(unittest.TestCase):
'LLM-Research/Phi-3-128k-instruct-GGUF')
self.assertTrue(template.template_type == TemplateType.phi3)
@unittest.skip#Unless(test_level() >= 0, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_load_ollama(self):
ollama = TemplateLoader.to_ollama(
'LLM-Research/Meta-Llama-3.1-8B-Instruct-GGUF')