phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 00:32:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refine

Browse source
This commit is contained in:
Botao Chen 2025-01-03 16:27:32 -08:00
parent 280581a4a3
commit bbe190a085
3 changed files with 23 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

1
llama_stack/apis/post_training/post_training.py
View file

@ -68,6 +68,7 @@ class TrainingConfig(BaseModel):
n_epochs: int
max_steps_per_epoch: int
gradient_accumulation_steps: int
max_validation_steps: int
data_config: DataConfig
optimizer_config: OptimizerConfig
efficiency_config: Optional[EfficiencyConfig] = None