add ResidualSimVQ to the special Sequential module

lucidrains · Nov 12, 2024 · f883a07 · f883a07
1 parent 81d0f3b
commit f883a07
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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 = "[email protected]" }

diff --git 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,