pin to version of vector quantize library that has backwards compatib…

…le interface back to codebook
lucidrains · Aug 8, 2022 · c8ad99d · c8ad99d
1 parent 82ab8d8
commit c8ad99d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/parti_pytorch/version.py b/parti_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.0.15'
+__version__ = '0.0.16'
diff --git a/setup.py b/setup.py
@@ -25,7 +25,7 @@
     'torch>=1.6',
     'torchvision',
     'transformers',
-    'vector-quantize-pytorch>=0.7.3'
+    'vector-quantize-pytorch>=0.9.2'
   ],
   classifiers=[
     'Development Status :: 4 - Beta',