mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
testing 3
This commit is contained in:
parent
dd7047dd5a
commit
8a9e6945ee
4 changed files with 18 additions and 1 deletions
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue