Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

remove llama2 dup in chat task

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/14359280
* remove llama2 dup in chat task
This commit is contained in:
mulin.lyh
2023-10-19 12:12:35 +08:00
parent f493ed007c
commit 8e187bdb96
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modelscope/models/nlp/llama/text_generation.py
View File

@@ -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):