depend on dalle2-pytorch for the CLIP and diffusion-prior

lucidrains · Sep 29, 2022 · 4260a32 · 4260a32
1 parent 92cad23
commit 4260a32
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -18,6 +18,7 @@
     'axial convolutions'
   ],
   install_requires=[
+    'dalle2-pytorch',
     'einops>=0.4',
     'torch>=1.6',
   ],