add validation to dpo_beta

connermanuel · connermanuel · commit 2cbd4c025e07 · 2025-05-08T14:25:01.000-07:00
diff --git a/src/together/resources/finetune.py b/src/together/resources/finetune.py
@@ -183,6 +183,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
`@@ -183,6 +183,9 @@ def create_finetune_request(`
`183`	`183`	`)`
`184`	`184`	`train_on_inputs = "auto"`
`185`	`185`
	`186`	`+ if dpo_beta is not None and training_method != "dpo":`
	`187`	`+ raise ValueError("dpo_beta is only supported for DPO training")`
	`188`	`+`
`186`	`189`	`lr_scheduler: FinetuneLRScheduler`
`187`	`190`	`if lr_scheduler_type == "cosine":`
`188`	`191`	`if scheduler_num_cycles <= 0.0:`