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 3 commits
  5. 05 Sep, 2022 1 commit
  6. 01 Sep, 2022 1 commit
  7. 31 Aug, 2022 2 commits
  8. 30 Aug, 2022 1 commit
  9. 22 Aug, 2022 2 commits
  10. 17 Aug, 2022 2 commits
  11. 16 Aug, 2022 2 commits
  12. 14 Aug, 2022 1 commit
  13. 21 Jul, 2022 1 commit
  14. 20 Jul, 2022 1 commit
  15. 29 Jun, 2022 1 commit
  16. 27 Jun, 2022 1 commit
  17. 22 Jun, 2022 1 commit
  18. 20 Jun, 2022 1 commit
  19. 17 Jun, 2022 2 commits
  20. 16 Jun, 2022 1 commit
  21. 12 Jun, 2022 1 commit
  22. 09 Jun, 2022 3 commits