src/diffusers/models/unets/unet_motion_model.py · 5266ab7935dd9e9aec596cdc2464badf1eacd99a · renzhc / diffusers_dcu

refactor: move model helper function in pipeline to a mixin class (#6571) · fa633ed6

elucida authored Feb 29, 2024



* move model helper function in pipeline to EfficiencyMixin

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

fa633ed6

unet_motion_model.py 41.2 KB

Replace unet_motion_model.py