using accelerate

Shoaibsi · Jan 6, 2023 · b67aaac · b67aaac
1 parent 2ca3e86
commit b67aaac
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -18,6 +18,7 @@
     'text-to-image'
   ],
   install_requires=[
+    'accelerate',
     'beartype',
     'einops>=0.6',
     'ema-pytorch',