diff --git a/pyproject.toml b/pyproject.toml index 17198628..3b682fbc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -59,7 +59,7 @@ bitsandbytes = "*" transformers = "4.44.2" einops-exts = "0.0.4" torchvision = "*" -accelerate = "0.33.0" +accelerate = "1.2.0" loguru = "*" vector-quantize-pytorch = "1.17.1" beartype = "0.18.5"