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-03 23:53:27 -08:00
parent 840fd353f7
commit 0b92fef3ba
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -43,7 +43,7 @@ class OptimizerConfig(BaseModel):
@json_schema_type @json_schema_type
class TrainingConfig(BaseModel): class TrainingConfig(BaseModel):
data_config: DataConfig data_config: DataConfig
optimizer_config: OptimizerConfig optimizer_config: Optional[OptimizerConfig] = OptimizerConfig()
n_epochs: Optional[int] = 1 n_epochs: Optional[int] = 1
max_steps_per_epoch: Optional[int] = None max_steps_per_epoch: Optional[int] = None
gradient_accumulation_steps: Optional[int] = 1 gradient_accumulation_steps: Optional[int] = 1
@ -159,8 +159,6 @@ class PostTraining(Protocol):
self, self,
job_uuid: str, job_uuid: str,
training_config: TrainingConfig, training_config: TrainingConfig,
hyperparam_search_config: Dict[str, Any],
logger_config: Dict[str, Any],
model: str = Field( model: str = Field(
default="Llama3.2-3B-Instruct", default="Llama3.2-3B-Instruct",
description="Model descriptor from `llama model list`", description="Model descriptor from `llama model list`",