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 · Oct 19, 2023 · 8e187bd · 8e187bd
1 parent f493ed0
commit 8e187bd
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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):