diff --git a/setup.py b/setup.py
index 0ae45559..1d453b2f 100644
--- 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',
+    ],
+)