ColossalAI/applications/ColossalMoE/colossal_moe/models
Hongxin Liu 65e5d6baa5 [moe] fix mixtral optim checkpoint (#5344) 2024-02-07 19:21:02 +08:00
..
__init__.py
mixtral_checkpoint.py [moe] fix mixtral optim checkpoint (#5344) 2024-02-07 19:21:02 +08:00
mixtral_layer.py [moe] support mixtral (#5309) 2024-02-07 19:21:02 +08:00
mixtral_policy.py [moe] fix mixtral forward default value (#5329) 2024-02-07 19:21:02 +08:00