Skip to content

GitLab

"...csrc/git@developer.sourcefind.cn:change/sglang.git" did not exist on "c480a3f6ea1be75d683551cdc7491aef2cf312e5"
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
convert_amused.py 27.1 KB