add validation to dpo_beta

connermanuel · connermanuel · commit 24c5938bcc95 · 2025-05-13T11:39:34.000-07:00
diff --git a/src/together/resources/finetune.py b/src/together/resources/finetune.py
@@ -175,6 +175,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
`@@ -175,6 +175,9 @@ def create_finetune_request(`
`175`	`175`	`)`
`176`	`176`	`train_on_inputs = "auto"`
`177`	`177`
	`178`	`+ if dpo_beta is not None and training_method != "dpo":`
	`179`	`+ raise ValueError("dpo_beta is only supported for DPO training")`
	`180`	`+`
`178`	`181`	`lr_scheduler: FinetuneLRScheduler`
`179`	`182`	`if lr_scheduler_type == "cosine":`
`180`	`183`	`if scheduler_num_cycles <= 0.0:`