diffusers
c5441965 - refactor: extract init/forward function in UNet2DConditionModel (#6478)

Commit
1 year ago
refactor: extract init/forward function in UNet2DConditionModel (#6478) * - extract function for stage in UNet2DConditionModel init & forward - Add new function get_mid_block() to unet_2d_blocks.py * add type hint to get_mid_block aligned with get_up_block and get_down_block; rename _set_xxx function * add type hint and use keyword arguments * remove `copy from` in versatile diffusion
Author
Parents
Loading