Add missing dependencies to setup.py

CompVis · Sep 28, 2021 · ee37e02 · ee37e02
1 parent d9b9ce0
commit ee37e02
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -9,5 +9,7 @@
         'torch',
         'numpy',
         'tqdm',
+        'pytorch-lightning',
+        'einops'
     ],
 )