PR #7083 Make mid block optional for flax UNet

Make mid block optional for flax UNet #7083

pcuenca merged 2 commits into huggingface:main from nyx-ai:flax-optional-mid_type

make mid block optional for flax UNet

ec4d3d6f

sayakpaul requested a review from

pcuenca 1 year ago

pcuenca approved these changes on 2024-03-07

make style

d8b904ab

pcuenca requested a review from

DN6 1 year ago

pcuenca merged b33b64f5 into main 1 year ago

Reviewers

pcuenca

DN6

Assignees

No one assigned

Labels

None yet

Milestone

No milestone