chore: Added default string to HuggingfacePostTrainingConfig

This commit is contained in:
sarthakdeshpande 2025-08-12 18:50:31 +05:30
parent 19123ca957
commit db6693ec21

View file

@ -71,7 +71,7 @@ class HuggingFacePostTrainingConfig(BaseModel):
dpo_beta: float = 0.1
use_reference_model: bool = True
dpo_loss_type: Literal["sigmoid", "hinge", "ipo", "kto_pair"] = "sigmoid"
dpo_output_dir: str
dpo_output_dir: str = ""
@classmethod
def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict[str, Any]: