From 441811ecd72bdd3ef66e9c6fe44e89c254d4c570 Mon Sep 17 00:00:00 2001 From: Zachary Mueller Date: Wed, 16 Nov 2022 15:57:07 -0500 Subject: [PATCH] Fix summarization script (#20286) --- examples/pytorch/summarization/run_summarization_no_trainer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/examples/pytorch/summarization/run_summarization_no_trainer.py b/examples/pytorch/summarization/run_summarization_no_trainer.py index 383143a6ef..d1752e3773 100644 --- a/examples/pytorch/summarization/run_summarization_no_trainer.py +++ b/examples/pytorch/summarization/run_summarization_no_trainer.py @@ -702,8 +702,6 @@ def main(): decoded_labels = tokenizer.batch_decode(labels, skip_special_tokens=True) decoded_preds, decoded_labels = postprocess_text(decoded_preds, decoded_labels) - - decoded_preds, decoded_labels = accelerator.gather_for_metrics(decoded_preds, decoded_labels) metric.add_batch( predictions=decoded_preds, references=decoded_labels,