refactor: extract init/forward function in UNet2DConditionModel#6478

Merged

patrickvonplaten merged 4 commits intohuggingface:mainfrom

ultranity:refactor_unet

Jan 19, 2024

Commits on Jan 7, 2024

- extract function for stage in UNet2DConditionModel init & forward
ultranity
committed

Commits on Jan 11, 2024

add type hint to get_mid_block aligned with get_up_block and get_down_block; rename _set_xxx function
ultranity
committed

Commits on Jan 16, 2024

add type hint and use keyword arguments
ultranity
committed

Commits on Jan 18, 2024

remove `copy from` in versatile diffusion
ultranity
committed