Files

NanoCode012 a1da39cd48 Feat(wandb): Refactor to be more flexible (#767 )

* Feat: Update to handle wandb env better

* chore: rename wandb_run_id to wandb_name

* feat: add new recommendation and update config

* fix: indent and pop disabled env if project passed

* feat: test env set for wandb and recommendation

* feat: update to use wandb_name and allow id

* chore: add info to readme

2023-12-04 22:17:25 +09:00

config.yml

Feat(wandb): Refactor to be more flexible (#767 )

2023-12-04 22:17:25 +09:00

README.md

address PR feedback

2023-06-10 14:23:56 -04:00

README.md

Pythia 12B

Single-GPU A100 only (?)

python scripts/finetune.py examples/pythia-12b/config.yml

⚠️ Multiple-GPU A100 - Doesn't seem to work with multi-gpu without causing OOM! ⚠️