Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • elucida's avatar
    refactor: extract init/forward function in UNet2DConditionModel (#6478) · c5441965
    elucida authored 
    * - 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
    c5441965
unet_2d_blocks.py 139 KB