riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Use attention_mask everywhere.

Browse Source
This commit is contained in:
Morgan Funtowicz 2019-12-09 14:13:17 +01:00
parent 348e19aa21
commit fe0f552e00
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
transformers/pipelines.py
View File

@ -154,9 +154,6 @@ class QuestionAnsweringPipeline(Pipeline):
return_attention_masks=True, return_input_lengths=False
)
# TODO : Harmonize model arguments across all model
inputs['attention_mask'] = inputs.pop('encoder_attention_mask')
if is_tf_available():
# TODO trace model
start, end = self.model(inputs)