src/diffusers/models/vae_flax.py · c01ec2d119b8d15d16c4fd6700f443c8512499bc · renzhc / diffusers_dcu

Suraj Patil authored Sep 20, 2022

* rename weights to align with PT

* DiagonalGaussianDistribution => FlaxDiagonalGaussianDistribution

* fix name

c01ec2d1

vae_flax.py 19.5 KB