Apply huggingface#24134 fix to cae78c4

zhanwenchen · Nov 1, 2023 · 5a27c37 · 5a27c37
1 parent cae78c4
commit 5a27c37
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
@@ -2753,6 +2753,7 @@ def save_model(self, output_dir: Optional[str] = None, _internal_call: bool = Fa
             or ShardedDDPOption.ZERO_DP_3 in self.args.sharded_ddp
             or self.fsdp is not None
         ):
+            os.makedirs(output_dir, exist_ok=True)
             state_dict = self.model.state_dict()
 
             if self.args.should_save: