From dde2a4cec687aacf24c38fac6c90dbc2ae9b37dd Mon Sep 17 00:00:00 2001 From: "zhangzhicheng.zzc" Date: Fri, 29 Jul 2022 22:14:52 +0800 Subject: [PATCH] [to #42322933]bug fixed for csanmnt Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9578933 --- modelscope/models/nlp/csanmt_for_translation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modelscope/models/nlp/csanmt_for_translation.py b/modelscope/models/nlp/csanmt_for_translation.py index 1af5b8ce..5ff24fea 100644 --- a/modelscope/models/nlp/csanmt_for_translation.py +++ b/modelscope/models/nlp/csanmt_for_translation.py @@ -18,13 +18,13 @@ __all__ = ['CsanmtForTranslation'] @MODELS.register_module(Tasks.translation, module_name=Models.translation) class CsanmtForTranslation(Model): - def __init__(self, model_dir, params, *args, **kwargs): + def __init__(self, model_dir, *args, **kwargs): """ Args: params (dict): the model configuration. """ super().__init__(model_dir, *args, **kwargs) - self.params = params + self.params = kwargs def forward(self, input: Dict[str, Tensor]) -> Dict[str, Tensor]: """return the result by the model