1. 06 Dec, 2023 3 commits
    • Pedro Cuenca's avatar
      Use CC12M for LCM WDS training example (#5908) · ab6672fe
      Pedro Cuenca authored
      * Fix SD scripts - there are only 2 items per batch
      
      * Adjustments to make the SDXL scripts work with other datasets
      
      * Use public webdataset dataset for examples
      
      * make style
      
      * Minor tweaks to the readmes.
      
      * Stress that the database is illustrative.
      ab6672fe
    • Dhruv Nair's avatar
      fix · f90a5139
      Dhruv Nair authored
      f90a5139
    • Sayak Paul's avatar
      [feat] allow SDXL pipeline to run with fused QKV projections (#6030) · a2bc2e14
      Sayak Paul authored
      
      
      * debug
      
      * from step
      
      * print
      
      * turn sigma a list
      
      * make str
      
      * init_noise_sigma
      
      * comment
      
      * remove prints
      
      * feat: introduce fused projections
      
      * change to a better name
      
      * no grad
      
      * device.
      
      * device
      
      * dtype
      
      * okay
      
      * print
      
      * more print
      
      * fix: unbind -> split
      
      * fix: qkv >-> k
      
      * enable disable
      
      * apply attention processor within the method
      
      * attn processors
      
      * _enable_fused_qkv_projections
      
      * remove print
      
      * add fused projection to vae
      
      * add todos.
      
      * add: documentation and cleanups.
      
      * add: test for qkv projection fusion.
      
      * relax assertions.
      
      * relax further
      
      * fix: docs
      
      * fix-copies
      
      * correct error message.
      
      * Empty-Commit
      
      * better conditioning on disable_fused_qkv_projections
      
      * check
      
      * check processor
      
      * bfloat16 computation.
      
      * check latent dtype
      
      * style
      
      * remove copy temporarily
      
      * cast latent to bfloat16
      
      * fix: vae -> self.vae
      
      * remove print.
      
      * add _change_to_group_norm_32
      
      * comment out stuff that didn't work
      
      * Apply suggestions from code review
      Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
      
      * reflect patrick's suggestions.
      
      * fix imports
      
      * fix: disable call.
      
      * fix more
      
      * fix device and dtype
      
      * fix conditions.
      
      * fix more
      
      * Apply suggestions from code review
      Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
      
      ---------
      Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
      a2bc2e14
  2. 05 Dec, 2023 7 commits
  3. 04 Dec, 2023 10 commits
  4. 02 Dec, 2023 3 commits
  5. 01 Dec, 2023 15 commits
  6. 30 Nov, 2023 1 commit
  7. 29 Nov, 2023 1 commit