From c293b8bc7031ea482f2f2836ce1b43e9e255af5a Mon Sep 17 00:00:00 2001 From: AllentDan Date: Thu, 7 Mar 2024 12:03:16 +0800 Subject: [PATCH] recover two names --- lmdeploy/cli/cli.py | 6 +++--- lmdeploy/model.py | 1 + 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/lmdeploy/cli/cli.py b/lmdeploy/cli/cli.py index c66aa76023..d29275a86e 100644 --- a/lmdeploy/cli/cli.py +++ b/lmdeploy/cli/cli.py @@ -110,11 +110,11 @@ def list(args): 'internlm2', 'internlm2-1_8b', 'internlm2-20b', 'internlm2-7b', 'internlm2-chat-1_8b', 'internlm2-chat-20b', 'internlm2-chat-7b', 'llama-2-chat', 'llama2', 'qwen-14b', 'qwen-7b', 'solar-70b', - 'yi-200k', 'yi-34b', 'yi-chat' + 'yi-200k', 'yi-34b', 'yi-chat', 'Mistral-7B-Instruct', + 'Mixtral-8x7B-Instruct' ] model_names = [ - n for n in model_names - if n.lower() not in deprecate_names + ['base'] + n for n in model_names if n not in deprecate_names + ['base'] ] model_names.sort() print('Supported model names:') diff --git a/lmdeploy/model.py b/lmdeploy/model.py index b38bcf7e62..a0dd2fa482 100644 --- a/lmdeploy/model.py +++ b/lmdeploy/model.py @@ -781,6 +781,7 @@ def match(cls, model_path: str) -> Optional[str]: @MODELS.register_module(name=['mistral-instruct', 'mixtral-instruct']) +@MODELS.register_module(name=['Mistral-7B-Instruct', 'Mixtral-8x7B-Instruct']) class MistralChat(BaseChatTemplate): """Template of Mistral and Mixtral Instruct models.