Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "81b707a449653c95301ca43a6d4875fbb70b5201"
Switch branch/tag
Find file BlameHistoryPermalink
  • Sayak Paul's avatar
    [Big refactor] move unets to `unets` module 🦋 (#6630) · 1f0705ad
    Sayak Paul authored 
    * move unets to  module 🦋

* parameterize unet-level import.

* fix flax unet2dcondition model import

* models __init__

* mildly depcrecating models.unet_2d_blocks in favor of models.unets.unet_2d_blocks.

* noqa

* correct depcrecation behaviour

* inherit from the actual classes.

* Empty-Commit

* backwards compatibility for unet_2d.py

* backward compatibility for unet_2d_condition

* bc for unet_1d

* bc for unet_1d_blocks
    1f0705ad
vae.py 35.8 KB