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

ray bugfix

Browse Source
This commit is contained in:
Dan Saunders
2025-03-31 18:35:41 +00:00
parent 32ce167404
commit 70b466aa67
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/train.py
View File

@@ -509,6 +509,7 @@ def train(
# Save the trained model and cleanup # Save the trained model and cleanup
save_trained_model(cfg, trainer, model, safe_serialization) save_trained_model(cfg, trainer, model, safe_serialization)
create_model_card(cfg, trainer) create_model_card(cfg, trainer)
cleanup_distributed() if not cfg.use_ray:
cleanup_distributed()
return model, tokenizer, trainer return model, tokenizer, trainer