use the checkpoint path as the temporary path for the Hugging Face tr…

…ainer.
TL-System · Nov 9, 2023 · e5092df · e5092df
1 parent b57ebc7
commit e5092df
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/plato/trainers/huggingface.py b/plato/trainers/huggingface.py
@@ -78,7 +78,10 @@ def __init__(self, model=None, callbacks=None):
 
         parser = HfArgumentParser(TrainingArguments)
         (self.training_args,) = parser.parse_args_into_dataclasses(
-            args=["--output_dir=/tmp", "--report_to=none"]
+            args=[
+                "--output_dir=" + Config.params["checkpoint_path"],
+                "--report_to=none",
+            ]
         )
 
         model_name = Config().trainer.model_name