chore: 更新项目名称

2026-03-31 09:34:11 +08:00
parent 780b9e1855
commit 2e009cf59a
51 changed files with 118 additions and 118 deletions
@@ -0,0 +1,10 @@
+from astrai.model.module import (
+    Linear,
+    RMSNorm,
+    MLP,
+    GQA,
+    DecoderBlock,
+)
+from astrai.model.transformer import Transformer
+
+__all__ = ["Linear", "RMSNorm", "MLP", "GQA", "DecoderBlock", "Transformer"]