Add dropout parameter to UNet2DModel/UNet2DConditionModel #4882
patrickvonplaten
merged
2 commits into
huggingface:main
from
dg845:unet-2d-model-expose-dropout-param
Add dropout param to get_down_block/get_up_block and UNet2DModel/UNet…
13b5a8cb
patrickvonplaten
approved these changes
on 2023-09-04
Add dropout param to Versatile Diffusion modeling, which has a copy o…
c146f87a
Login to write a write a comment.
Login via GitHub