Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 04:29:22 +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
committed by wenmeng.zwm
parent 2c3bf9629d
commit 0390bdea67
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):