diff --git a/audiolm_pytorch/version.py b/audiolm_pytorch/version.py
index 04188a1..36a511e 100644
--- a/audiolm_pytorch/version.py
+++ b/audiolm_pytorch/version.py
@@ -1 +1 @@
-__version__ = '2.2.0'
+__version__ = '2.2.1'
diff --git a/setup.py b/setup.py
index d724580..b6f5b60 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,7 @@
     'torchaudio',
     'transformers',
     'tqdm',
-    'vector-quantize-pytorch>=1.18.8'
+    'vector-quantize-pytorch>=1.19.2'
   ],
   classifiers=[
     'Development Status :: 4 - Beta',