examples/controlnet/train_controlnet_sd3.py · 8ead643bb786fe6bc80c9a4bd1730372d410a9df · renzhc / diffusers_dcu

[examples/controlnet/train_controlnet_sd3.py] Fixes #11050 - Cast... · 8ead643b

Andreas Jörg authored Mar 14, 2025


[examples/controlnet/train_controlnet_sd3.py] Fixes #11050 - Cast prompt_embeds and pooled_prompt_embeds to weight_dtype to prevent dtype mismatch (#11051)

Fix: dtype mismatch of prompt embeddings in sd3 controlnet training
Co-authored-by: Andreas Jörg <andreasjoerg@MacBook-Pro-von-Andreas-2.fritz.box>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

8ead643b

train_controlnet_sd3.py 56.3 KB

Replace train_controlnet_sd3.py