From 30695154b4202248b6141cb3019461b1da5ddd9d Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Sat, 12 Apr 2025 21:44:33 +0200 Subject: [PATCH] removed depcreated models from init files --- src/aiia/__init__.py | 2 +- src/aiia/model/__init__.py | 5 ----- 2 files changed, 1 insertion(+), 6 deletions(-) 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