diffusers
refactor: extract init/forward function in UNet2DConditionModel
#6478

Merged

refactor: extract init/forward function in UNet2DConditionModel #6478

patrickvonplaten merged 4 commits into huggingface:main from ultranity:refactor_unet

- extract function for stage in UNet2DConditionModel init & forward

135f57bd

patrickvonplaten commented on 2024-01-09

patrickvonplaten requested a review from

DN6 1 year ago

patrickvonplaten requested a review from

yiyixuxu 1 year ago

patrickvonplaten requested a review from

sayakpaul 1 year ago

yiyixuxu commented on 2024-01-09

sayakpaul commented on 2024-01-11

DN6 commented on 2024-01-11

add type hint to get_mid_block aligned with get_up_block and get_down…

4001c86f

ultranity changed the title ~~[WIP] refactor: extract init/forward function in UNet2DConditionModel~~ refactor: extract init/forward function in UNet2DConditionModel 1 year ago

patrickvonplaten commented on 2024-01-15

patrickvonplaten approved these changes on 2024-01-15

add type hint and use keyword arguments

f899f925

ultranity force pushed from 84a6de60 to f899f925 1 year ago

patrickvonplaten commented on 2024-01-17

remove `copy from` in versatile diffusion

6e37ae36

patrickvonplaten merged c5441965 into main 1 year ago

ultranity deleted the refactor_unet branch 1 year ago

yiyixuxu added refactor

yiyixuxu added Good Example PR

Reviewers

patrickvonplaten

DN6

sayakpaul

yiyixuxu

Assignees

No one assigned

Labels

Good Example PR refactor

Milestone

No milestone

diffusers refactor: extract init/forward function in UNet2DConditionModel #6478 Merged

refactor: extract init/forward function in UNet2DConditionModel #6478

diffusers
refactor: extract init/forward function in UNet2DConditionModel
#6478

Merged