"""VQ-VAE module.""" from .vqvae import VQVAE