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

revert a mis modification

Browse Source
This commit is contained in:
雨泓
2022-06-23 11:20:54 +08:00
parent 75dd131ada
commit 2eb633ec93
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
modelscope/models/nlp/masked_language_model.py
View File

@@ -31,20 +31,20 @@ class MaskedLMModelBase(Model):
return self.model.config
return None
def forward(self, inputs: Dict[str, Tensor]) -> Dict[str, np.ndarray]:
def forward(self, input: Dict[str, Tensor]) -> Dict[str, np.ndarray]:
"""return the result by the model
Args:
inputs (Dict[str, Any]): the preprocessed data
input (Dict[str, Any]): the preprocessed data
Returns:
Dict[str, np.ndarray]: results
"""
rst = self.model(
input_ids=inputs['input_ids'],
attention_mask=inputs['attention_mask'],
token_type_ids=inputs['token_type_ids'])
return {'logits': rst['logits'], 'input_ids': inputs['input_ids']}
input_ids=input['input_ids'],
attention_mask=input['attention_mask'],
token_type_ids=input['token_type_ids'])
return {'logits': rst['logits'], 'input_ids': input['input_ids']}
@MODELS.register_module(Tasks.fill_mask, module_name=Models.structbert)