fix: 移除多余 request 参数并增强 tokenizer 健壮性

- 路由和 _get_engine 不再需要 request 参数,直接引用模块级 app
- from_pretrained 增加文件完整性校验,缺 tokenizer.json 则抛 FileNotFoundError
- 移除 from_pretrained 中未使用的 **kwargs
This commit is contained in:
2026-05-17 12:52:18 +08:00
parent c241a5dcef
commit 8f1b32f2b6
3 changed files with 55 additions and 36 deletions
+7 -5
View File
@@ -16,9 +16,13 @@ def required(**kw):
@dataclass
class TrainConfig(BaseConfig):
# basic setting
model: nn.Module = field(default=None, metadata=required(help="Model for training."))
model: nn.Module = field(
default=None, metadata=required(help="Model for training.")
)
strategy: str = field(default=None, metadata=required(help="Training strategy."))
dataset: Dataset = field(default=None, metadata=required(help="Dataset for training."))
dataset: Dataset = field(
default=None, metadata=required(help="Dataset for training.")
)
optimizer_fn: Callable[[nn.Module], Optimizer] = field(
default=None, metadata=required(help="Optimizer factory for training.")
)
@@ -99,6 +103,4 @@ class TrainConfig(BaseConfig):
def validate(self):
for fld in fields(self):
if fld.metadata.get("required") and getattr(self, fld.name) is None:
raise ValueError(
f"TrainConfig.{fld.name} is required but got None."
)
raise ValueError(f"TrainConfig.{fld.name} is required but got None.")