Fix the bug of sd3 controlnet training when using gradient checkpointing.#9498

Merged

yiyixuxu merged 2 commits intohuggingface:mainfrom pibbo88:sd3_controlnet_train_grad_ckpt_fix

Sep 23, 2024

+1-1

Commits on Sep 23, 2024

Fix the bug of sd3 controlnet training when using gradient_checkpointing. Refer to issue huggingface#9496
pibbo88
committed
Merge branch 'main' into sd3_controlnet_train_grad_ckpt_fix
pibbo88
authored