diff --git a/pyproject.toml b/pyproject.toml
index 4740a4c1..e95ecc32 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,7 +32,7 @@ datasets = "*"
 lion-pytorch = "*"
 sentencepiece = "*"
 colt5-attention = "0.10.14"
-vector-quantize-pytorch = "1.1.5"
+vector-quantize-pytorch = "1.8.1"
 tokenmonster = "*"
 scipy = "*"
 beartype = "*"