diffusers
feat: add `Dropout` to Flax UNet
#3894
Merged

Commits
  • feat: add Dropout to Flax UNet
    SauravMaheshkar committed 2 years ago
  • Merge branch 'main' into saurav/add-dropout-flax-unet
    SauravMaheshkar committed 2 years ago
  • feat: add @compact decorator
    SauravMaheshkar committed 2 years ago
  • Merge branch 'main' into saurav/add-dropout-flax-unet
    SauravMaheshkar committed 2 years ago
  • Merge branch 'main' into saurav/add-dropout-flax-unet
    SauravMaheshkar committed 2 years ago
  • Merge branch 'main' into saurav/add-dropout-flax-unet
    SauravMaheshkar committed 2 years ago
  • Merge branch 'main' into saurav/add-dropout-flax-unet
    SauravMaheshkar committed 2 years ago
  • Merge branch 'main' into saurav/add-dropout-flax-unet
    SauravMaheshkar committed 2 years ago
  • Merge branch 'main' into saurav/add-dropout-flax-unet
    SauravMaheshkar committed 2 years ago
  • Merge branch 'main' into saurav/add-dropout-flax-unet
    SauravMaheshkar committed 2 years ago
  • Merge branch 'main' into saurav/add-dropout-flax-unet
    SauravMaheshkar committed 2 years ago
  • fix: drop nn.compact
    SauravMaheshkar committed 2 years ago
Loading