From 357141f6de171db6f2d65fd66f950896207a0e7b Mon Sep 17 00:00:00 2001 From: Botao Chen Date: Sun, 9 Mar 2025 18:14:26 -0700 Subject: [PATCH] refine --- llama_stack/apis/post_training/post_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/apis/post_training/post_training.py b/llama_stack/apis/post_training/post_training.py index 8cbc66daf..134b6bc15 100644 --- a/llama_stack/apis/post_training/post_training.py +++ b/llama_stack/apis/post_training/post_training.py @@ -19,7 +19,7 @@ from llama_stack.schema_utils import json_schema_type, register_schema, webmetho @json_schema_type class TrainingStrategy(BaseModel): # 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 num_warmup_steps: Optional[Union[int, Literal["auto"]]] = "auto"