diff --git a/src/aiia/__init__.py b/src/aiia/__init__.py
index 838c765..a4e0d44 100644
--- a/src/aiia/__init__.py
+++ b/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
diff --git a/src/aiia/model/__init__.py b/src/aiia/model/__init__.py
index a45512a..88f3211 100644
--- a/src/aiia/model/__init__.py
+++ b/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",
-
 ]
\ No newline at end of file