[Community] Add dropout to Flax UNet

Open patrickvonplaten opened this issue 3 years ago • 3 comments

Describe the bug

Flax UNet currently doesn't support dropout

Reproduction

No response

Logs

No response

System Info

Main

Sep 15 '22 14:09 patrickvonplaten

Hello @patrickvonplaten !

I assume this issue is still relevant? If so happy to help with it. Just for clarity: the task is to address the comments that patil had here: https://github.com/huggingface/diffusers/pull/502#pullrequestreview-1109050359, correct? i.e. that the flax UNet model does not actually perform dropout since it was originally intended just for inference?

Nov 01 '22 21:11 MarkRich

Hey @MarkRich,

That's exactly right. We simply need to add dropout keywords to a lot of places so that it can be customized by the user

Nov 02 '22 17:11 patrickvonplaten

I won't have time to look into this anytime soon, so unassigning from me.

Dec 20 '22 00:12 patrickvonplaten