diff --git a/requirements.txt b/requirements.txt index e094a665..5cfa2b44 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,7 +10,7 @@ datasets torchfix torchdiffeq>=0.2.3,<0.3.0 beartype>=0.15.0,<0.16.0 -vector-quantize-pytorch>=1.12.0,<1.13.0 +vector-quantize-pytorch>=1.12.0,<1.28.0 loguru rich==13.7.1 tiktoken==0.6.0