Files
ColossalAI/colossalai/booster/plugin/dp_plugin_base.py
Hongxin Liu 6c0fa7b9a8 [llama] fix dataloader for hybrid parallel (#5358)
* [plugin] refactor prepare dataloader

* [plugin] update train script
2024-02-05 15:14:56 +08:00

3.0 KiB