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

temp commit

Browse source
This commit is contained in:
Botao Chen 2024-11-27 14:35:01 -08:00
parent c31a78dfcb
commit 18ae577760
2 changed files with 21 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -183,16 +183,16 @@ class PostTraining(Protocol):
@webmethod(route="/post-training/supervised-fine-tune")
def supervised_fine_tune(
self,
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],
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]],
) -> PostTrainingJob: ...
@webmethod(route="/post-training/preference-optimize")