diff --git a/setup.py b/setup.py
index 53ee988..546e9aa 100644
--- a/setup.py
+++ b/setup.py
@@ -22,6 +22,7 @@
         "transformers",
         "DeepCache==0.1.1",
         "torch",
+        "accelerate",
     ],
     python_requires='>=3.8.0',
 )