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

change output of siamese_uie to dict

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11666360
This commit is contained in:
fubang.zfb
2023-02-15 04:18:21 +00:00
committed by wenmeng.zwm
parent 2738f659cf
commit fafb9cadfd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/pipelines/nlp/siamese_uie_pipeline.py
View File

@@ -120,7 +120,7 @@ class SiameseUiePipeline(Pipeline):
prefix_info = []
self.forward(text, tokenized_text, prefix_info, schema, pred_info_list,
output_all_prefix)
return pred_info_list
return {'output': pred_info_list}
def _pad(self, input_ids, pad_token_id):
input_ids[-1] += [pad_token_id] * (self.max_len - len(input_ids[-1]))