From 1bfef605907f818253e4a5e96b3a5d0645c3e9b0 Mon Sep 17 00:00:00 2001 From: Botao Chen Date: Fri, 13 Dec 2024 16:12:56 -0800 Subject: [PATCH] refine --- llama_stack/apis/post_training/post_training.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/llama_stack/apis/post_training/post_training.py b/llama_stack/apis/post_training/post_training.py index ff8edd412..fdbaa364d 100644 --- a/llama_stack/apis/post_training/post_training.py +++ b/llama_stack/apis/post_training/post_training.py @@ -169,6 +169,12 @@ class PostTraining(Protocol): training_config: TrainingConfig, hyperparam_search_config: Dict[str, Any], logger_config: Dict[str, Any], + model: str = Field( + default="Llama3.2-3B-Instruct", + description="Model descriptor from `llama model list`", + ), + checkpoint_dir: Optional[str] = None, + algorithm_config: Optional[AlgorithmConfig] = None, ) -> PostTrainingJob: ... @webmethod(route="/post-training/preference-optimize", method="POST")