diff --git a/setup.py b/setup.py
index 8e36652..21ed312 100644
--- a/setup.py
+++ b/setup.py
@@ -19,6 +19,7 @@
   ],
   install_requires=[
     'accelerate',
+    'diffusers',
     'datasets',
     'beartype',
     'einops>=0.6',