diff --git a/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py b/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py index 32bce9531b71..89a2aa2e83bd 100644 --- a/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py +++ b/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py @@ -355,7 +355,7 @@ def parse_args(input_args=None): help="A prompt that is used during validation to verify that the model is learning.", ) parser.add_argument( - "--validation_image", + "--validation_image_path", type=str, default=None, help="path to an image that is used during validation as the condition image to verify that the model is learning.",