diff --git a/pyproject.toml b/pyproject.toml index 1f404343..39e73670 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,7 +28,7 @@ typing = "3.7.4.3" transformers = "4.36.0" einops-exts = "0.0.4" torchvision = "*" -accelerate = "0.22.0" +accelerate = "0.25.0" datasets = "2.10.1" lion-pytorch = "0.0.7" sentencepiece = "0.1.99"