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

Fix check for falsey inputs in run_summarization (#18155)

Browse Source
This commit is contained in:
John Giorgi 2022-07-18 00:50:32 -07:00 committed by GitHub
parent ccc0897804
commit c46d39f390
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/summarization/run_summarization.py
View File

@ -515,7 +515,7 @@ def main():
inputs, targets = [], [] inputs, targets = [], []
for i in range(len(examples[text_column])): for i in range(len(examples[text_column])):
if examples[text_column][i] is not None and examples[summary_column][i] is not None: if examples[text_column][i] and examples[summary_column][i]:
inputs.append(examples[text_column][i]) inputs.append(examples[text_column][i])
targets.append(examples[summary_column][i]) targets.append(examples[summary_column][i])