refactor: sort out dependencies

Co-Authored-By: oc013 <101832295+oc013@users.noreply.github.com>
Co-Authored-By: Aarni Koskela <akx@iki.fi>
Co-Authored-By: oc013 <101832295+oc013@users.noreply.github.com>
Co-Authored-By: Aarni Koskela <akx@iki.fi>

environment.yaml

@@ -3,40 +3,41 @@ channels:
   - pytorch
   - defaults
 dependencies:
+  - cudatoolkit=11.3
   - git
-  - python=3.8.5
+  - numpy=1.22.3
   - pip=20.3
-  - cudatoolkit=11.3
+  - python=3.8.5
   - pytorch=1.11.0
+  - scikit-image=0.19.2
   - torchvision=0.12.0
-  - numpy=1.19.2
   - pip:
+    - -e .
+    - -e git+https://github.com/CompVis/taming-transformers#egg=taming-transformers
+    - -e git+https://github.com/openai/CLIP#egg=clip
+    - -e git+https://github.com/TencentARC/GFPGAN#egg=GFPGAN
+    - -e git+https://github.com/xinntao/Real-ESRGAN#egg=realesrgan
+    - -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion
+    - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion
+    - accelerate==0.12.0
     - albumentations==0.4.3
-    - opencv-python==4.1.2.30
-    - opencv-python-headless==4.1.2.30
-    - pudb==2019.2
-    - imageio==2.9.0
-    - imageio-ffmpeg==0.4.2
-    - pytorch-lightning==1.4.2
-    - omegaconf==2.1.1
-    - test-tube>=0.7.5
+    - basicsr>=1.3.4.0
+    - diffusers<=0.2.4
     - einops==0.3.0
-    - torch-fidelity==0.3.0
-    - transformers==4.19.2
-    - torchmetrics==0.6.0
-    - kornia==0.6
+    - facexlib>=0.2.3
     - gradio==3.1.6
-    - accelerate==0.12.0
+    - imageio-ffmpeg==0.4.2
+    - imageio==2.9.0
+    - kornia==0.6
+    - omegaconf==2.1.1
+    - opencv-python-headless==4.6.0.66
+    - pudb==2019.2
     - pynvml==11.4.1
-    - basicsr>=1.3.4.0
-    - facexlib>=0.2.3
     - python-slugify>=6.1.2
-    - streamlit>=1.12.2
+    - pytorch-lightning==1.4.2
     - retry>=0.9.2
-    - diffusers<=0.2.4
-    - -e git+https://github.com/CompVis/taming-transformers#egg=taming-transformers
-    - -e git+https://github.com/openai/CLIP#egg=clip
-    - -e git+https://github.com/TencentARC/GFPGAN#egg=GFPGAN
-    - -e git+https://github.com/xinntao/Real-ESRGAN#egg=realesrgan
-    - -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion
-    - -e .
+    - streamlit>=1.12.2
+    - test-tube>=0.7.5
+    - torch-fidelity==0.3.0
+    - torchmetrics==0.6.0
+    - transformers==4.19.2

setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 setup(
-    name='latent-diffusion',
+    name='sd-webui',
     version='0.0.1',
     description='',
     packages=find_packages(),