src/diffusers/models/unets/unet_2d_condition.py · 63b684684979b22d9cfbeacc534c2f274e95e3ab · renzhc / diffusers_dcu

[LoRA] pop the LoRA scale so that it doesn't get propagated to the weeds (#7338) · ce9825b5

Sayak Paul authored Mar 19, 2024



* pop scale from the top-level unet instead of getting it.

* improve readability.

* Apply suggestions from code review
Co-authored-by: YiYi Xu <yixu310@gmail.com>

* fix a little bit.

---------
Co-authored-by: YiYi Xu <yixu310@gmail.com>

ce9825b5

unet_2d_condition.py 65.4 KB

Replace unet_2d_condition.py