From 1fcb31b0443b0bc6651edad3f20d7ef70c0d7186 Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Wed, 26 Mar 2025 21:25:52 +0100
Subject: [PATCH] removed aiia from being able to get called

---
 src/aiia/model/__init__.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/aiia/model/__init__.py b/src/aiia/model/__init__.py
index f68a42a..c0ad6e7 100644
--- a/src/aiia/model/__init__.py
+++ b/src/aiia/model/__init__.py
@@ -1,21 +1,21 @@
 from .Model import (
-    AIIA,
     AIIABase,
     AIIABaseShared,
     AIIAchunked,
     AIIAExpert,
     AIIAmoe,
-    AIIArecursive
+    AIIArecursive,
+    AIIASparseMoe
 )
 from .config import AIIAConfig
 
 __all__ = [
-    "AIIA",
     "AIIABase",
     "AIIABaseShared",
     "AIIAchunked",
     "AIIAExpert",
     "AIIAmoe",
     "AIIArecursive",
-    "AIIAConfig"
+    "AIIAConfig",
+    "AIIASparseMoe"
 ]
\ No newline at end of file