from .quantizer import VectorQuantizer, DiagonalGaussianDistribution