add validation to dpo_beta

connermanuel · connermanuel · commit bddb4d8405a2 · 2025-04-17T11:23:05.000-07:00
diff --git a/src/together/resources/finetune.py b/src/together/resources/finetune.py
@@ -171,6 +171,9 @@ def create_finetune_request(
         )
         train_on_inputs = "auto"
 
+    if dpo_beta is not None and training_method != "dpo":
+        raise ValueError("dpo_beta is only supported for DPO training")
+
     lr_scheduler: FinetuneLRScheduler
     if lr_scheduler_type == "cosine":
         if scheduler_num_cycles <= 0.0:

Original file line number	Diff line number	Diff line change
`@@ -171,6 +171,9 @@ def create_finetune_request(`
`171`	`171`	`)`
`172`	`172`	`train_on_inputs = "auto"`
`173`	`173`
	`174`	`+ if dpo_beta is not None and training_method != "dpo":`
	`175`	`+ raise ValueError("dpo_beta is only supported for DPO training")`
	`176`	`+`
`174`	`177`	`lr_scheduler: FinetuneLRScheduler`
`175`	`178`	`if lr_scheduler_type == "cosine":`
`176`	`179`	`if scheduler_num_cycles <= 0.0:`