fix (#1678)

huggingface · May 29, 2024 · 488b502 · 488b502
1 parent 3c0a10b
commit 488b502
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/trl/trainer/sft_trainer.py b/trl/trainer/sft_trainer.py
@@ -141,6 +141,11 @@ def __init__(
         dataset_kwargs: Optional[Dict] = None,
         eval_packing: Optional[bool] = None,
     ):
+        if args is None:
+            output_dir = "tmp_trainer"
+            warnings.warn(f"No `SFTConfig` passed, using `output_dir={output_dir}`.")
+            args = SFTConfig(output_dir=output_dir)
+
         if model_init_kwargs is not None:
             warnings.warn(
                 "You passed `model_init_kwargs` to the SFTTrainer, the value you passed will override the one in the `SFTConfig`."