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:
-  - git
-  - python=3.8.5
-  - pip=20.3
   - cudatoolkit=11.3
+  - git
+  - numpy=1.22.3
+  - pip=20.3
+  - python=3.8.5
   - pytorch=1.11.0
+  - scikit-image=0.19.2
   - torchvision=0.12.0
-  - numpy=1.19.2
   - pip:
-    - 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
-    - einops==0.3.0
-    - torch-fidelity==0.3.0
-    - transformers==4.19.2
-    - torchmetrics==0.6.0
-    - kornia==0.6
-    - gradio==3.1.6
-    - accelerate==0.12.0
-    - pynvml==11.4.1
-    - basicsr>=1.3.4.0
-    - facexlib>=0.2.3
-    - python-slugify>=6.1.2
-    - streamlit>=1.12.2
-    - retry>=0.9.2
-    - diffusers<=0.2.4
+    - -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 .
+    - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion
+    - accelerate==0.12.0
+    - albumentations==0.4.3
+    - basicsr>=1.3.4.0
+    - diffusers<=0.2.4
+    - einops==0.3.0
+    - facexlib>=0.2.3
+    - gradio==3.1.6
+    - 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
+    - python-slugify>=6.1.2
+    - pytorch-lightning==1.4.2
+    - retry>=0.9.2
+    - 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(),