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

updated init to remove expeort model and move the smoe higher
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Failing after 31s Details

This commit is contained in:
Falko Victor Habel 2025-03-26 21:57:17 +01:00
parent 458d78a23e
commit 5987a130f6
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/aiia/__init__.py
View File

@ -1,4 +1,4 @@
from .model.Model import AIIABase, AIIABaseShared, AIIAchunked, AIIAExpert, AIIAmoe, AIIArecursive, AIIASparseMoe from .model.Model import AIIABase, AIIABaseShared, AIIAchunked, AIIAmoe, AIIASparseMoe, AIIArecursive
from .model.config import AIIAConfig from .model.config import AIIAConfig
from .data.DataLoader import DataLoader from .data.DataLoader import DataLoader
from .pretrain.pretrainer import Pretrainer, ProjectionHead from .pretrain.pretrainer import Pretrainer, ProjectionHead

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

@ -2,10 +2,9 @@ from .Model import (
AIIABase, AIIABase,
AIIABaseShared, AIIABaseShared,
AIIAchunked, AIIAchunked,
AIIAExpert,
AIIAmoe, AIIAmoe,
AIIArecursive, AIIASparseMoe,
AIIASparseMoe AIIArecursive
) )
from .config import AIIAConfig from .config import AIIAConfig
@ -13,9 +12,9 @@ __all__ = [
"AIIABase", "AIIABase",
"AIIABaseShared", "AIIABaseShared",
"AIIAchunked", "AIIAchunked",
"AIIAExpert",
"AIIAmoe", "AIIAmoe",
"AIIASparseMoe",
"AIIArecursive", "AIIArecursive",
"AIIAConfig", "AIIAConfig",
"AIIASparseMoe"
] ]