gumbel_vector_quantizer.py 6.74 KB