1. 21 Sep, 2022 1 commit
    • Pedro Cuenca's avatar
      Allow dtype to be specified in Flax pipeline (#600) · fb2fbab1
      Pedro Cuenca authored
      * Fix typo in docstring.
      
      * Allow dtype to be overridden on model load.
      
      This may be a temporary solution until #567 is addressed.
      
      * Create latents in float32
      
      The denoising loop always computes the next step in float32, so this
      would fail when using `bfloat16`.
      fb2fbab1
  2. 19 Sep, 2022 1 commit
  3. 17 Sep, 2022 1 commit
  4. 16 Sep, 2022 1 commit
  5. 14 Sep, 2022 1 commit
  6. 13 Sep, 2022 1 commit
  7. 08 Sep, 2022 1 commit
  8. 01 Sep, 2022 1 commit
  9. 17 Aug, 2022 1 commit
  10. 21 Jul, 2022 1 commit
  11. 20 Jul, 2022 2 commits
  12. 19 Jul, 2022 2 commits
  13. 15 Jul, 2022 1 commit
  14. 14 Jul, 2022 1 commit
  15. 20 Jun, 2022 1 commit
  16. 17 Jun, 2022 3 commits
  17. 15 Jun, 2022 1 commit
  18. 14 Jun, 2022 2 commits
  19. 13 Jun, 2022 1 commit
  20. 12 Jun, 2022 1 commit
  21. 09 Jun, 2022 3 commits
  22. 08 Jun, 2022 1 commit
  23. 07 Jun, 2022 3 commits
  24. 06 Jun, 2022 1 commit
  25. 02 Jun, 2022 1 commit
  26. 01 Jun, 2022 1 commit
  27. 31 May, 2022 1 commit