Skip to content

GitLab

"benchmarks/vscode:/vscode.git/clone" did not exist on "5de4edd1fa9ed144c8218415be85677297bbb494"
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_blocks.py 122 KB