src/aiia/__init__.py

@@ -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 .data.DataLoader import DataLoader
 from .pretrain.pretrainer import Pretrainer, ProjectionHead

src/aiia/model/__init__.py

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