fix

sayakpaul · sayakpaul · commit 73e9d054b23d · 2025-07-07T19:53:02.000+05:30
diff --git a/examples/dreambooth/train_dreambooth_lora_flux_kontext.py b/examples/dreambooth/train_dreambooth_lora_flux_kontext.py
@@ -312,7 +312,7 @@ def parse_args(input_args=None):
     parser.add_argument(
         "--cond_image_column",
         type=str,
-        default="image",
+        default=None,
         help="Column in the dataset containing the condition image. Must be specified when performing I2I fine-tuning",
     )
     parser.add_argument(

Original file line number	Diff line number	Diff line change
`@@ -312,7 +312,7 @@ def parse_args(input_args=None):`
`312`	`312`	`parser.add_argument(`
`313`	`313`	`"--cond_image_column",`
`314`	`314`	`type=str,`
`315`		`- default="image",`
	`315`	`+ default=None,`
`316`	`316`	`help="Column in the dataset containing the condition image. Must be specified when performing I2I fine-tuning",`
`317`	`317`	`)`
`318`	`318`	`parser.add_argument(`