src/diffusers/models/unet_2d_blocks.py · dd4459ad795230ee6d70dc270484bb9b523db9d4 · renzhc / diffusers_dcu

[Refactor] spliting`ResnetBlock2D` into multiple blocks (#6166) · dd4459ad

YiYi Xu authored Jan 09, 2024





---------
Co-authored-by: yiyixuxu <yixu310@gmail,com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

dd4459ad

unet_2d_blocks.py 136 KB

Replace unet_2d_blocks.py