mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-15 06:00:48 +00:00
chore: Added default string to HuggingfacePostTrainingConfig
This commit is contained in:
parent
19123ca957
commit
db6693ec21
1 changed files with 1 additions and 1 deletions
|
@ -71,7 +71,7 @@ class HuggingFacePostTrainingConfig(BaseModel):
|
||||||
dpo_beta: float = 0.1
|
dpo_beta: float = 0.1
|
||||||
use_reference_model: bool = True
|
use_reference_model: bool = True
|
||||||
dpo_loss_type: Literal["sigmoid", "hinge", "ipo", "kto_pair"] = "sigmoid"
|
dpo_loss_type: Literal["sigmoid", "hinge", "ipo", "kto_pair"] = "sigmoid"
|
||||||
dpo_output_dir: str
|
dpo_output_dir: str = ""
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict[str, Any]:
|
def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict[str, Any]:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue