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

Comment changes are shownComment changes are hidden
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
  • src/diffusers
    • models
      • File
        unet_2d_blocks.py
      • File
        unet_2d_condition.py
    • pipelines/deprecated/versatile_diffusion
      • File
        modeling_text_unet.py