diff --git a/pyproject.toml b/pyproject.toml index c5435eb5..7cc1063d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,7 +30,7 @@ bitsandbytes = "*" transformers = "4.42.3" einops-exts = "0.0.4" torchvision = "*" -accelerate = "0.31.0" +accelerate = "0.33.0" datasets = "*" loguru = "*" vector-quantize-pytorch = "1.14.7"