Skip to content

GitLab

Unverified Commit 97ec17f7 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

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

parent b513ec8b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
src/transformers/trainer.py
View file @ 97ec17f7
...@@ -2631,6 +2631,9 @@ class Trainer: ...@@ -2631,6 +2631,9 @@ class Trainer:
dataset: Optional[Union[str, List[str]]] = None, dataset: Optional[Union[str, List[str]]] = None,
dataset_args: 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( training_summary = TrainingSummary.from_trainer(
self, self,
language=language, language=language,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment