Fix accelerate version

2023-09-15 22:54:51 -07:00
parent 500edd2b53
commit b590ec0a7c
1 changed files with 1 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@@ -9,6 +9,7 @@ setup(
        "ml-collections",
        "absl-py",
        "diffusers[torch]==0.17.1",
+        "accelerate==0.17",
        "wandb",
        "torchvision",
        "inflect==6.0.4",