fix ray train and add fsdp2 smoke test for ray trainer (#3053)

* add fsdp2 smokle test for ray trainer * fix raytrain with fsdp2
2025-08-11 09:31:54 -04:00
parent 9b12c05660
commit d4d84d48af
2 changed files with 76 additions and 1 deletions
--- a/src/axolotl/cli/utils/train.py
+++ b/src/axolotl/cli/utils/train.py
@@ -123,6 +123,9 @@ def launch_training(
            _launch_torchrun_training(cfg_file, kwargs, launcher_args, use_exec)
        elif launcher == "python":
            _launch_python_training(cfg_file, kwargs)
+    elif launcher is None:
+        # handle ray train launch
+        _launch_python_training(cfg_file, kwargs)


 def _launch_cloud_training(