search

foundation-model-stack / fms-fsdp

🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash attention v2.
https://pytorch.org/docs/stable/fsdp.html
Apache License 2.0
116 stars 18 forks source link

Clean up training configs #14

Closed lchu-ibm closed 4 months ago

lchu-ibm commented 4 months ago

fix https://github.com/foundation-model-stack/fms-fsdp/issues/7

lchu-ibm commented 4 months ago

lint didn't pass, will push a quick fix.