diffusers
03d829d5 - feat: add `Dropout` to Flax UNet (#3894)

Commit

2 years ago

feat: add `Dropout` to Flax UNet (#3894) * feat: add Dropout to Flax UNet * feat: add @compact decorator * fix: drop nn.compact

References

Author

SauravMaheshkar

Parents