Flux2 dreambooth: AttributeError: 'Namespace' object has no attribute 'validation_image_path'

[hannalaguilar]

Describe the bug

In train_dreambooth_lora_flux2_img2img.py on line 1422, there is a variable named validation_image_path, but shouldn’t it actually be validation_image?

validation_image = load_image(args.validation_image_path).convert("RGB") AttributeError: 'Namespace' object has no attribute 'validation_image_path'

Reproduction

accelerate launch train_dreambooth_lora_flux2_img2img.py \
  --pretrained_model_name_or_path=black-forest-labs/FLUX.2-dev  \
  --output_dir="flux2-i2i" \
  --dataset_name="kontext-community/relighting" \
  --image_column="output" --cond_image_column="file_name" --caption_column="instruction" \
  --do_fp8_training \
  --gradient_checkpointing \
  --remote_text_encoder \
  --cache_latents \
  --resolution=1024 \
  --train_batch_size=1 \
  --guidance_scale=1 \
  --gradient_accumulation_steps=4 \
  --gradient_checkpointing \
  --optimizer="adamw" \
  --use_8bit_adam \
  --cache_latents \
  --learning_rate=1e-4 \
  --lr_scheduler="constant_with_warmup" \
  --lr_warmup_steps=200 \
  --max_train_steps=1000 \
  --rank=16\
  --seed="0" 

Logs

System Info

diffusers 0.36.0.dev0

Who can help?

No response