From c8ad99d0b0a0b42c7172ba954c0961fbbc2b0790 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Sun, 7 Aug 2022 20:53:33 -0700
Subject: [PATCH] pin to version of vector quantize library that has backwards
 compatible interface back to codebook

---
 parti_pytorch/version.py | 2 +-
 setup.py                 | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/parti_pytorch/version.py b/parti_pytorch/version.py
index 8ecd910..19c6940 100644
--- 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
index 26b7d78..345504d 100644
--- 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',