vector_quantize_pytorch