diff --git a/pyproject.toml b/pyproject.toml index b99426dd..ee2039cd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -25,7 +25,7 @@ bitsandbytes = "0.43.0" transformers = "4.41.0" einops-exts = "0.0.4" torchvision = "0.18.0" -accelerate = "0.28.0" +accelerate = "0.30.1" datasets = "*" loguru = "*" vector-quantize-pytorch = "1.14.7"