chore: 优化未使用的模块

2026-04-06 09:54:17 +08:00
parent a57a16430d
commit ace8f6ee68
5 changed files with 24 additions and 165 deletions
@@ -12,18 +12,19 @@ from astrai.inference import (
    InferenceEngine,
 )
 from astrai.model import AutoModel, Transformer
-from astrai.tokenize import BpeTokenizer
-from astrai.trainer import SchedulerFactory, StrategyFactory, Trainer
+from astrai.tokenize import AutoTokenizer
+from astrai.trainer import CallbackFactory, SchedulerFactory, StrategyFactory, Trainer

 __all__ = [
    "Transformer",
    "ModelConfig",
    "TrainConfig",
    "DatasetFactory",
-    "BpeTokenizer",
+    "AutoTokenizer",
    "GenerationRequest",
    "InferenceEngine",
    "Trainer",
+    "CallbackFactory",
    "StrategyFactory",
    "SchedulerFactory",
    "BaseFactory",