setup.py

fix for stable diffusion
Birch-san · Nov 5, 2022 · 0602097 · 0602097
1 parent 60e5042
commit 0602097
Showing 1 changed file with 23 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -1,5 +1,25 @@
-from setuptools import setup
+from setuptools import setup, find_packages
 
 
-if __name__ == '__main__':
-    setup()
+setup(
+    name='k-diffusion',
+    version='0.0.1',
+    description='Karras et al. (2022) diffusion models for PyTorch',
+    packages=find_packages(),
+    install_requires=[
+        'accelerate',
+        'clean-fid',
+        'einops',
+        'jsonmerge',
+        'kornia',
+        'Pillow',
+        'resize-right',
+        'scikit-image',
+        'scipy',
+        'torch',
+        'torchdiffeq',
+        'torchvision',
+        'tqdm',
+        'wandb',
+    ],
+)