phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

refine

Browse source
This commit is contained in:
Botao Chen 2025-03-09 18:14:26 -07:00
parent 41f86b5ce6
commit 357141f6de
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -19,7 +19,7 @@ from llama_stack.schema_utils import json_schema_type, register_schema, webmetho
@json_schema_type @json_schema_type
class TrainingStrategy(BaseModel): class TrainingStrategy(BaseModel):
# params that control Optimizer # params that control Optimizer
lr: Optional[Union[float, Literal["auto"]]] = "auto" learning_rate: Optional[Union[float, Literal["auto"]]] = "auto"
weight_decay: Optional[float] = 0.1 weight_decay: Optional[float] = 0.1
num_warmup_steps: Optional[Union[int, Literal["auto"]]] = "auto" num_warmup_steps: Optional[Union[int, Literal["auto"]]] = "auto"