feat: 实现模型动态注册机制

2026-04-05 19:38:12 +08:00
parent ff43a2fab8
commit fc278d17ab
25 changed files with 686 additions and 651 deletions
@@ -6,5 +6,17 @@ from astrai.model.module import (
    RMSNorm,
 )
 from astrai.model.transformer import Transformer
+from astrai.model.automodel import AutoModel

-__all__ = ["Linear", "RMSNorm", "MLP", "GQA", "DecoderBlock", "Transformer"]
+
+__all__ = [
+    # Modules
+    "Linear",
+    "RMSNorm",
+    "MLP",
+    "GQA",
+    "DecoderBlock",
+    # Models
+    "Transformer",
+    "AutoModel",
+]