Commits · 8685699392cfaa79c9c9ab0e16fca49b0f0574f3 · renzhc / diffusers_dcu · GitLab

20 Sep, 2022 1 commit

[FlaxAutoencoderKL] rename weights to align with PT (#584) · c01ec2d1

Suraj Patil authored Sep 20, 2022

* rename weights to align with PT

* DiagonalGaussianDistribution => FlaxDiagonalGaussianDistribution

* fix name

c01ec2d1

19 Sep, 2022 2 commits
- [Flax] Solve problem with VAE (#574) · 2a8477de
  Patrick von Platen authored Sep 19, 2022
  
  2a8477de
- [Flax] Add Vae for Stable Diffusion (#555) · bf5ca036
  Patrick von Platen authored Sep 19, 2022
```
* [Flax] Add Vae

* correct

* Apply suggestions from code review
Co-authored-by: Suraj Patil <surajp815@gmail.com>

* Finish
Co-authored-by: Suraj Patil <surajp815@gmail.com>
```
  bf5ca036