diffusers
55e17907 - Add dropout parameter to UNet2DModel/UNet2DConditionModel (#4882)

Commit
2 years ago
Add dropout parameter to UNet2DModel/UNet2DConditionModel (#4882) * Add dropout param to get_down_block/get_up_block and UNet2DModel/UNet2DConditionModel. * Add dropout param to Versatile Diffusion modeling, which has a copy of UNet2DConditionModel and its own get_down_block/get_up_block functions.
Author
Parents
Loading