diff --git a/pyproject.toml b/pyproject.toml index 096d697..671334e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "vector-quantize-pytorch" -version = "1.20.5" +version = "1.20.7" description = "Vector Quantization - Pytorch" authors = [ { name = "Phil Wang", email = "lucidrains@gmail.com" } diff --git a/vector_quantize_pytorch/utils.py b/vector_quantize_pytorch/utils.py index d591a09..8ab0fcc 100644 --- a/vector_quantize_pytorch/utils.py +++ b/vector_quantize_pytorch/utils.py @@ -13,6 +13,7 @@ from vector_quantize_pytorch.residual_fsq import ResidualFSQ, GroupedResidualFSQ from vector_quantize_pytorch.latent_quantization import LatentQuantize from vector_quantize_pytorch.sim_vq import SimVQ +from vector_quantize_pytorch.residual_sim_vq import ResidualSimVQ QUANTIZE_KLASSES = ( VectorQuantize, @@ -22,6 +23,7 @@ FSQ, LFQ, SimVQ, + ResidualSimVQ, ResidualLFQ, GroupedResidualLFQ, ResidualFSQ,