testing 3

This commit is contained in:
Jai Suphavadeeprasit 2026-02-02 13:08:42 -05:00
parent dd7047dd5a
commit 8a9e6945ee
4 changed files with 18 additions and 1 deletions

View file

@ -49,6 +49,13 @@ class TrainingConfig(BaseModel):
"trained_model_checkpoints",
description="Base path to save model checkpoints"
)
checkpoint_interval: int = Field(
3,
description=(
"Save checkpoint every N training steps. "
"Set to 0 to only save final checkpoint."
),
)
# === vLLM Server Configuration ===
vllm_restart_interval: int = Field(