1. 15 Sep, 2022 1 commit
    • Kashif Rasul's avatar
      Karras VE, DDIM and DDPM flax schedulers (#508) · b34be039
      Kashif Rasul authored
      * beta never changes removed from state
      
      * fix typos in docs
      
      * removed unused var
      
      * initial ddim flax scheduler
      
      * import
      
      * added dummy objects
      
      * fix style
      
      * fix typo
      
      * docs
      
      * fix typo in comment
      
      * set return type
      
      * added flax ddom
      
      * fix style
      
      * remake
      
      * pass PRNG key as argument and split before use
      
      * fix doc string
      
      * use config
      
      * added flax Karras VE scheduler
      
      * make style
      
      * fix dummy
      
      * fix ndarray type annotation
      
      * replace returns a new state
      
      * added lms_discrete scheduler
      
      * use self.config
      
      * add_noise needs state
      
      * use config
      
      * use config
      
      * docstring
      
      * added flax score sde ve
      
      * fix imports
      
      * fix typos
      b34be039
  2. 14 Sep, 2022 1 commit
  3. 13 Sep, 2022 2 commits
  4. 08 Sep, 2022 4 commits
  5. 07 Sep, 2022 1 commit
  6. 05 Sep, 2022 1 commit
  7. 01 Sep, 2022 1 commit
  8. 30 Aug, 2022 1 commit
  9. 22 Aug, 2022 2 commits
  10. 16 Aug, 2022 5 commits
  11. 14 Aug, 2022 1 commit
  12. 09 Aug, 2022 1 commit
  13. 28 Jul, 2022 1 commit
  14. 27 Jul, 2022 1 commit
  15. 22 Jul, 2022 1 commit
  16. 21 Jul, 2022 3 commits
  17. 20 Jul, 2022 2 commits
  18. 19 Jul, 2022 2 commits
  19. 13 Jul, 2022 1 commit
  20. 12 Jul, 2022 1 commit
  21. 29 Jun, 2022 1 commit
  22. 28 Jun, 2022 1 commit
  23. 26 Jun, 2022 1 commit
  24. 25 Jun, 2022 2 commits
  25. 24 Jun, 2022 1 commit
  26. 22 Jun, 2022 1 commit