diffusers
bc6e1f24 - Fix incorrect batch temporal IDs for `cond_model_input` in Flux2 Klein img2img training (#13923)

Commit
33 days ago
Fix incorrect batch temporal IDs for `cond_model_input` in Flux2 Klein img2img training (#13923) `train_dreambooth_lora_flux2_klein_img2img.py` passed the whole batch to `Flux2KleinPipeline._prepare_image_ids` as a list of reference images, so each independent training sample got a different temporal id (T=10, 20, 30, ...) as if they were multiple references of one instance. Each batch element is an independent sample with a single conditional image, so they must share the same temporal id. Generate ids for one sample and expand across the batch. This is the Flux2 Klein counterpart of the same bug/fix in `train_dreambooth_lora_flux2_img2img.py` (#13811). Fixes #13866
Author
Parents
Loading