fix ut of checking template name (#1382)

InternLM · Apr 2, 2024 · 70c0b1f · 70c0b1f
1 parent 9b8ebc1
commit 70c0b1f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_lmdeploy/test_model.py b/tests/test_lmdeploy/test_model.py
@@ -18,7 +18,8 @@
      ('WizardLM/WizardLM-70B-V1.0', ['wizardlm']),
      ('codellama/CodeLlama-34b-Instruct-hf', ['codellama']),
      ('deepseek-ai/deepseek-coder-6.7b-instruct', ['deepseek-coder']),
-     ('deepseek-ai/deepseek-vl-7b-chat', ['deepseek', 'deepseek-chat']),
+     ('deepseek-ai/deepseek-vl-7b-chat', ['deepseek-vl']),
+     ('deepseek-ai/deepseek-moe-16b-chat', ['deepseek']),
      ('tiiuae/falcon-7b', ['falcon']), ('workspace', [None])])
 @pytest.mark.parametrize('suffix', ['', '-w4', '-4bit', '-16bit'])
 def test_best_match_model(model_path_and_name, suffix):