From 0390bdea676a1804ddd7099b6d8aef57a695fda2 Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Thu, 19 Oct 2023 12:12:35 +0800 Subject: [PATCH] remove llama2 dup in chat task Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/14359280 * remove llama2 dup in chat task --- modelscope/models/nlp/llama/text_generation.py | 1 - 1 file changed, 1 deletion(-) diff --git a/modelscope/models/nlp/llama/text_generation.py b/modelscope/models/nlp/llama/text_generation.py index d95cae34..45b9d5f0 100644 --- a/modelscope/models/nlp/llama/text_generation.py +++ b/modelscope/models/nlp/llama/text_generation.py @@ -74,7 +74,6 @@ def get_chat_prompt(system: str, text: str, history: List[Tuple[str, str]], @MODELS.register_module(Tasks.chat, module_name=Models.llama2) @MODELS.register_module(Tasks.chat, module_name=Models.llama) @MODELS.register_module(Tasks.text_generation, module_name=Models.llama2) -@MODELS.register_module(Tasks.chat, module_name=Models.llama2) @MODELS.register_module(Tasks.text_generation, module_name=Models.llama) class LlamaForTextGeneration(MsModelMixin, LlamaForCausalLM, TorchModel):