Standardize out_dir behavior

Lightning-AI / litgpt

Pretrain, finetune, deploy 20+ LLMs on your own data. Uses state-of-the-art techniques: flash attention, FSDP, 4-bit, LoRA, and more.

https://lightning.ai

Apache License 2.0

6.85k stars 726 forks source link

Closed rasbt closed 3 weeks ago

rasbt commented 3 weeks ago

Apply the init_out_dir(out_dir) used for pretraining also to the finetuning scripts.

Fixes #1312

awaelchli commented 3 weeks ago

The test_init_out_dir in test_pretrain.py should now be moved to test_utils.py as well.

rasbt commented 3 weeks ago

The test_init_out_dir in test_pretrain.py should now be moved to test_utils.py as well.

Good catch, thanks!

rasbt commented 3 weeks ago

I think the failing tests are because HF is down again ...