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

feat/tf_support #37

Merged
Fabel merged 13 commits from feat/tf_support into develop 2025-04-16 20:59:48 +00:00
Conversation 0 Commits 13 Files Changed 13 +1 -6
2 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 30695154b4 - Show all commits

2
src/aiia/__init__.py
View File

@ -1,4 +1,4 @@
from .model.Model import AIIABase, AIIABaseShared, AIIAchunked, AIIAmoe, AIIASparseMoe, AIIArecursive from .model.Model import AIIABase, AIIABaseShared, AIIAmoe, AIIASparseMoe
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

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

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