Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • dg845's avatar
    Add dropout parameter to UNet2DModel/UNet2DConditionModel (#4882) · 55e17907
    dg845 authored 
    * 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.
    55e17907
unet_2d_condition.py 52.6 KB