1. 03 Dec, 2025 1 commit
  2. 30 Sep, 2025 1 commit
  3. 19 Jun, 2025 1 commit
  4. 24 Feb, 2025 1 commit
  5. 28 Jan, 2025 1 commit
    • Aryan's avatar
      Refactor gradient checkpointing (#10611) · c4d4ac21
      Aryan authored
      * update
      
      * remove unused fn
      
      * apply suggestions based on review
      
      * update + cleanup 🧹
      
      * more cleanup 🧹
      
      * make fix-copies
      
      * update test
      c4d4ac21
  6. 05 Dec, 2024 1 commit
  7. 02 Dec, 2024 1 commit
  8. 27 Nov, 2024 1 commit
  9. 20 Nov, 2024 1 commit
  10. 08 Nov, 2024 1 commit
  11. 07 Nov, 2024 1 commit
    • Sayak Paul's avatar
      [Core] introduce `controlnet` module (#8768) · ded3db16
      Sayak Paul authored
      
      
      * move vae flax module.
      
      * controlnet module.
      
      * prepare for PR.
      
      * revert a commit
      
      * gracefully deprecate controlnet deps.
      
      * fix
      
      * fix doc path
      
      * fix-copies
      
      * fix path
      
      * style
      
      * style
      
      * conflicts
      
      * fix
      
      * fix-copies
      
      * sparsectrl.
      
      * updates
      
      * fix
      
      * updates
      
      * updates
      
      * updates
      
      * fix
      
      ---------
      Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
      ded3db16
  12. 23 Sep, 2024 1 commit
  13. 09 Sep, 2024 1 commit
  14. 13 Aug, 2024 1 commit
  15. 24 Jul, 2024 1 commit
    • Sayak Paul's avatar
      [Core] fix QKV fusion for attention (#8829) · 50d21f7c
      Sayak Paul authored
      * start debugging the problem,
      
      * start
      
      * fix
      
      * fix
      
      * fix imports.
      
      * handle hunyuan
      
      * remove residuals.
      
      * add a check for making sure there's appropriate procs.
      
      * add more rigor to the tests.
      
      * fix test
      
      * remove redundant check
      
      * fix-copies
      
      * move check_qkv_fusion_matches_attn_procs_length and check_qkv_fusion_processors_exist.
      50d21f7c
  16. 08 Jul, 2024 1 commit
  17. 02 Jul, 2024 1 commit
  18. 01 Jul, 2024 1 commit
  19. 26 Jun, 2024 1 commit
  20. 21 Jun, 2024 1 commit
  21. 19 Jun, 2024 1 commit
  22. 18 Jun, 2024 1 commit
  23. 12 Jun, 2024 2 commits