diff --git a/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py b/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py index f53a28bb34fa..f4b79e452338 100644 --- a/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py +++ b/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py @@ -1483,7 +1483,7 @@ def _encode_single(prompt: str): ) if args.validation_prompt is not None: - validation_image = load_image(args.validation_image_path).convert("RGB") + validation_image = load_image(args.validation_image).convert("RGB") validation_kwargs = {"image": validation_image} if args.remote_text_encoder: validation_kwargs["prompt_embeds"] = compute_remote_text_embeddings(args.validation_prompt)