rosinality / vq-vae-2-pytorch Python 1.8K 279 Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch vq-vae-2 vq-vae