[Refactor] Use small bloom model in model partitioning's test

xrsrke / pipegoose

Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*

MIT License

76 stars 17 forks source link

Closed xrsrke closed 9 months ago