1. 24 Sep, 2022 1 commit
  2. 22 Sep, 2022 1 commit
    • Suraj Patil's avatar
      [UNet2DConditionModel] add gradient checkpointing (#461) · e7120bae
      Suraj Patil authored
      * add grad ckpt to downsample blocks
      
      * make it work
      
      * don't pass gradient_checkpointing to upsample block
      
      * add tests for UNet2DConditionModel
      
      * add test_gradient_checkpointing
      
      * add gradient_checkpointing for up and down blocks
      
      * add functions to enable and disable grad ckpt
      
      * remove the forward argument
      
      * better naming
      
      * make supports_gradient_checkpointing private
      e7120bae
  3. 16 Sep, 2022 1 commit
  4. 08 Sep, 2022 2 commits
  5. 07 Sep, 2022 1 commit
  6. 05 Sep, 2022 2 commits
  7. 30 Aug, 2022 1 commit
  8. 19 Aug, 2022 1 commit
  9. 18 Aug, 2022 1 commit
  10. 14 Aug, 2022 1 commit
  11. 13 Aug, 2022 1 commit
  12. 12 Aug, 2022 1 commit
  13. 11 Aug, 2022 1 commit
  14. 09 Aug, 2022 1 commit
  15. 20 Jul, 2022 5 commits
  16. 19 Jul, 2022 3 commits
  17. 18 Jul, 2022 1 commit
  18. 15 Jul, 2022 1 commit
  19. 29 Jun, 2022 1 commit
  20. 28 Jun, 2022 1 commit
  21. 27 Jun, 2022 4 commits
  22. 22 Jun, 2022 1 commit
  23. 17 Jun, 2022 2 commits
  24. 15 Jun, 2022 5 commits