1. 31 Jan, 2024 1 commit
    • Dhruv Nair's avatar
      Add PIA Model/Pipeline (#6698) · 56bea6b4
      Dhruv Nair authored
      * update
      
      * update
      
      * updaet
      
      * add tests and docs
      
      * clean up
      
      * add to toctree
      
      * fix copies
      
      * pr review feedback
      
      * fix copies
      
      * fix tests
      
      * update docs
      
      * update
      
      * update
      
      * update docs
      
      * update
      
      * update
      
      * update
      
      * update
      56bea6b4
  2. 29 Jan, 2024 1 commit
  3. 23 Jan, 2024 1 commit
    • 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