train_controlnet_sd3.py] Fixes #11050 - Cast prompt_embeds and pooled_prompt_embeds to weight_dtype to prevent dtype mismatch
#11051

Merged

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

sayakpaul merged 2 commits into huggingface:main from andjoer:fix-dtype-mismatch

Fix: dtype mismatch of prompt embeddings in sd3 controlnet training

2ec9ffa6

sayakpaul approved these changes on 2025-03-14

Merge branch 'main' into fix-dtype-mismatch

2c91843d

sayakpaul merged 8ead643b into main 278 days ago

Reviewers

sayakpaul

Assignees

No one assigned

Labels

None yet

Milestone

No milestone