ColossalAI/colossalai/moe
digger yu 5e1c93d732
[hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335)
Co-authored-by: binmakeswell <binmakeswell@gmail.com>
2024-03-05 21:52:30 +08:00
..
__init__.py [hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335) 2024-03-05 21:52:30 +08:00
_operation.py [moe] support mixtral (#5309) 2024-02-07 19:21:02 +08:00
checkpoint.py [hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335) 2024-03-05 21:52:30 +08:00
experts.py
layers.py
load_balance.py
loss.py
manager.py
routers.py [moe] fix tests 2024-02-08 12:46:37 +08:00
utils.py