7bef20f
1
2
3
from .quantizer import VectorQuantizer, DiagonalGaussianDistribution from .mvq import VectorQuantizerMVQ from .softvq import SoftVectorQuantizer