Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

feat/smoe #34

Merged
Fabel merged 7 commits from feat/smoe into main 2025-03-28 15:48:40 +00:00
Conversation 0 Commits 7 Files Changed 7 +4 -4
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 1fcb31b044 - Show all commits

8
src/aiia/model/__init__.py
View File

@ -1,21 +1,21 @@
from .Model import ( from .Model import (
AIIA,
AIIABase, AIIABase,
AIIABaseShared, AIIABaseShared,
AIIAchunked, AIIAchunked,
AIIAExpert, AIIAExpert,
AIIAmoe, AIIAmoe,
AIIArecursive AIIArecursive,
AIIASparseMoe
) )
from .config import AIIAConfig from .config import AIIAConfig
__all__ = [ __all__ = [
"AIIA",
"AIIABase", "AIIABase",
"AIIABaseShared", "AIIABaseShared",
"AIIAchunked", "AIIAchunked",
"AIIAExpert", "AIIAExpert",
"AIIAmoe", "AIIAmoe",
"AIIArecursive", "AIIArecursive",
"AIIAConfig" "AIIAConfig",
"AIIASparseMoe"
] ]