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

[to #42322933]bug fixed for csanmnt

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9578933
This commit is contained in:
zhangzhicheng.zzc
2022-07-29 22:14:52 +08:00
committed by yingda.chen
parent b96338996c
commit dde2a4cec6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/models/nlp/csanmt_for_translation.py
View File

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