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

temp commit

Browse source
This commit is contained in:
Botao Chen 2024-11-27 16:46:29 -08:00
parent bfc782c054
commit 6c709abc4d
3 changed files with 28 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -180,16 +180,16 @@ class PostTraining(Protocol):
@webmethod(route="/post-training/supervised-fine-tune")
def supervised_fine_tune(
self,
job_uuid: Optional[str],
model: Optional[str],
dataset_id: Optional[str],
validation_dataset_id: Optional[str],
algorithm: Optional[FinetuningAlgorithm],
algorithm_config: Optional[LoraFinetuningConfig],
optimizer_config: Optional[OptimizerConfig],
training_config: Optional[TrainingConfig],
hyperparam_search_config: Optional[Dict[str, Any]],
logger_config: Optional[Dict[str, Any]],
job_uuid: str,
model: str,
dataset_id: str,
validation_dataset_id: str,
algorithm: FinetuningAlgorithm,
algorithm_config: LoraFinetuningConfig,
optimizer_config: OptimizerConfig,
training_config: TrainingConfig,
hyperparam_search_config: Dict[str, Any],
logger_config: Dict[str, Any],
) -> PostTrainingJob: ...
@webmethod(route="/post-training/preference-optimize")