github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-18 16:00:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335)

Browse Source 
Co-authored-by: binmakeswell <binmakeswell@gmail.com>
This commit is contained in:
digger yu
2024-03-05 21:52:30 +08:00
committed by GitHub
parent a7ae2b5b4c
commit 5e1c93d732
7 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
colossalai/moe/__init__.py
View File

@@ -1,4 +1,4 @@
from .checkpoint import MoECheckpintIO
from .checkpoint import MoECheckpointIO
from .experts import MLPExperts
from .layers import SparseMLP, apply_load_balance
from .manager import MOE_MANAGER
@@ -14,7 +14,7 @@ __all__ = [
"NormalNoiseGenerator",
"UniformNoiseGenerator",
"SparseMLP",
"MoECheckpintIO",
"MoECheckpointIO",
"MOE_MANAGER",
"apply_load_balance",
]