ColossalAI/applications/ColossalMoE/colossal_moe/models
Hongxin Liu b60be18dcc [moe] fix mixtral checkpoint io (#5314) 2024-02-07 19:21:02 +08:00
..
__init__.py [moe] init mixtral impl 2024-02-07 19:21:02 +08:00
mixtral_checkpoint.py [moe] fix mixtral checkpoint io (#5314) 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] support mixtral (#5309) 2024-02-07 19:21:02 +08:00