bes-dev / vqvae_dwt_distiller.pytorch

Apache License 2.0
26 stars 1 forks source link

vqvae_dwt_distiller.pytorch

Simple improvement of VQVAE that allow to generate x2 sized images compared to baseline. It allows to generate 512x512 images using ruDALL-E.

POC checkpoint: https://drive.google.com/file/d/1GjGXs1l0mOiFxKJwutjTQyCHEaF-wrIL/view?usp=sharing