ColossalAI/applications/ColossalMoE/colossal_moe
Hongxin Liu 65e5d6baa5 [moe] fix mixtral optim checkpoint (#5344) 2024-02-07 19:21:02 +08:00
..
models [moe] fix mixtral optim checkpoint (#5344) 2024-02-07 19:21:02 +08:00
__init__.py [moe] init mixtral impl 2024-02-07 19:21:02 +08:00
utils.py [moe] support mixtral (#5309) 2024-02-07 19:21:02 +08:00