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

This commit is contained in:
2026-04-05 19:38:12 +08:00
parent ff43a2fab8
commit fc278d17ab
25 changed files with 686 additions and 651 deletions
+2 -1
View File
@@ -12,7 +12,7 @@ from astrai.inference import (
GenerationRequest,
InferenceEngine,
)
from astrai.model.transformer import Transformer
from astrai.model import AutoModel, Transformer
from astrai.trainer import SchedulerFactory, StrategyFactory, Trainer
__all__ = [
@@ -27,4 +27,5 @@ __all__ = [
"StrategyFactory",
"SchedulerFactory",
"BaseFactory",
"AutoModel",
]