tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix Ray train crashing after succeeding (#3542) [skip ci]

Browse Source
This commit is contained in:
Matthew Hambrecht
2026-03-25 07:38:28 -04:00
committed by GitHub
parent c2bd75aff6
commit 678ebb1bb2
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/axolotl/cli/train.py
View File

@@ -84,8 +84,11 @@ def do_cli(config: Union[Path, str] = Path("examples/"), **kwargs):
storage_path=Path(parsed_cfg.output_dir).absolute().as_posix(), storage_path=Path(parsed_cfg.output_dir).absolute().as_posix(),
), ),
) )
return trainer.fit()
return do_train(parsed_cfg, parsed_cli_args) trainer.fit()
return
do_train(parsed_cfg, parsed_cli_args)
def ray_train_func(kwargs: dict): def ray_train_func(kwargs: dict):