Fix gather for metrics (#19360)

ad98642a · Zachary Mueller · GitHub · d9101b71 · ad98642a
Unverified Commit ad98642a authored Oct 05, 2022 by Zachary Mueller Committed by GitHub Oct 05, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/summarization/run_summarization_no_trainer.py ...les/pytorch/summarization/run_summarization_no_trainer.py +1 -1

No files found.
--- a/examples/pytorch/summarization/run_summarization_no_trainer.py
+++ b/examples/pytorch/summarization/run_summarization_no_trainer.py
@@ -685,7 +685,7 @@ def main():
                    # If we did not pad to max length, we need to pad the labels too
                    labels = accelerator.pad_across_processes(batch["labels"], dim=1, pad_index=tokenizer.pad_token_id)

-                generated_tokens, labels = accelerator.gather_for_metrics(generated_tokens, labels)
+                generated_tokens, labels = accelerator.gather_for_metrics((generated_tokens, labels))
                generated_tokens = generated_tokens.cpu().numpy()
                labels = labels.cpu().numpy()