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

fix misc

Browse source
This commit is contained in:
Botao Chen 2024-12-13 14:55:01 -08:00
parent d55a8343ea
commit d0a72cc288
3 changed files with 7 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

@ -66,6 +66,7 @@ class TrainingConfig(BaseModel):
@json_schema_type
class LoraFinetuningConfig(BaseModel):
type: Literal["LoRA"] = "LoRA"
lora_attn_modules: List[str]
apply_lora_to_mlp: bool
apply_lora_to_output: bool
@ -77,12 +78,13 @@ class LoraFinetuningConfig(BaseModel):
@json_schema_type
class QATFinetuningConfig(BaseModel):
type: Literal["QAT"] = "QAT"
quantizer_name: str
group_size: int
AlgorithmConfig = Annotated[
Union[LoraFinetuningConfig, LoraFinetuningConfig], Field(discriminator="type")
Union[LoraFinetuningConfig, QATFinetuningConfig], Field(discriminator="type")
]