removed depcreated models from init files
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Failing after 0s Details

This commit is contained in:
Falko Victor Habel 2025-04-12 21:44:33 +02:00
parent d4db1ef116
commit 30695154b4
2 changed files with 1 additions and 6 deletions

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

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",
] ]