Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "4bd2952a376a7781526ffb063536b90e38b09573"
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_condition.py 65.9 KB