Only create the model card on process 0 (#14857)

97ec17f7 · Sylvain Gugger · GitHub · b513ec8b · 97ec17f7
Unverified Commit 97ec17f7 authored Dec 21, 2021 by Sylvain Gugger Committed by GitHub Dec 21, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/transformers/trainer.py src/transformers/trainer.py +3 -0

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -2631,6 +2631,9 @@ class Trainer:
        dataset: Optional[Union[str, List[str]]] = None,
        dataset_args: Optional[Union[str, List[str]]] = None,
    ):
+        if not self.is_world_process_zero():
+            return
        training_summary = TrainingSummary.from_trainer(
            self,
            language=language,