[ray] specify ray storage path (#6920)

Former-commit-id: 4be6b66b1eaa79955e936ce2b747a8837ecd1e49
2025-02-14 05:55:41 -08:00
parent a893505924
commit 76f9bd1820
2 changed files with 6 additions and 2 deletions
--- a/src/llamafactory/hparams/training_args.py
+++ b/src/llamafactory/hparams/training_args.py
@@ -16,7 +16,11 @@ class RayArguments:

    ray_run_name: Optional[str] = field(
        default=None,
-        metadata={"help": "The training results will be saved at `saves/ray_run_name`."},
+        metadata={"help": "The training results will be saved at `<ray_storage_path>/ray_run_name`."},
+    )
+    ray_storage_path: str = field(
+        default="./saves",
+        metadata={"help": "The storage path to save training results to"},
    )
    ray_num_workers: int = field(
        default=1,