diff --git a/pyproject.toml b/pyproject.toml
index fdffbfd3..9077ad2f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -33,7 +33,7 @@ datasets = "*"
 lion-pytorch = "*"
 sentencepiece = "*"
 colt5-attention = "0.10.14"
-vector-quantize-pytorch = "1.1.5"
+vector-quantize-pytorch = "1.7.1"
 tokenmonster = "*"
 scipy = "*"
 beartype = "*"