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,35 @@
+__version__ = "1.3.2"
+__author__ = "ViperEkura"
+
+from astrai.config import (
+    ModelConfig,
+    TrainConfig,
+)
+from astrai.model.transformer import Transformer
+from astrai.data import DatasetLoader, BpeTokenizer
+from astrai.inference.generator import (
+    GenerationRequest,
+    LoopGenerator,
+    StreamGenerator,
+    BatchGenerator,
+    EmbeddingEncoder,
+    GeneratorFactory,
+)
+from astrai.trainer import Trainer, StrategyFactory, SchedulerFactory
+
+__all__ = [
+    "Transformer",
+    "ModelConfig",
+    "TrainConfig",
+    "DatasetLoader",
+    "BpeTokenizer",
+    "GenerationRequest",
+    "LoopGenerator",
+    "StreamGenerator",
+    "BatchGenerator",
+    "EmbeddingEncoder",
+    "GeneratorFactory",
+    "Trainer",
+    "StrategyFactory",
+    "SchedulerFactory",
+]