from .vq_base import VectorQuantizer