src/diffusers/models/vae.py · f085d2f5c6569a1c0d90327c51328622036ef76e · renzhc / diffusers_dcu

[Type Hint] VAE models (#365) · f085d2f5

Partho authored Sep 05, 2022



* [Type Hint] VAE models

* Update src/diffusers/models/vae.py

* apply suggestions from code review
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>

f085d2f5

vae.py 17.3 KB

Replace vae.py