diff --git a/setup.py b/setup.py
index 98eb2a9..6e51703 100644
--- 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",