From 8039021b53042d26011b461f66e6f00efbd78771 Mon Sep 17 00:00:00 2001 From: Thomas Mello Date: Sun, 11 Sep 2022 04:18:15 +0300 Subject: [PATCH] refactor: sort out dependencies Co-Authored-By: oc013 <101832295+oc013@users.noreply.github.com> Co-Authored-By: Aarni Koskela Co-Authored-By: oc013 <101832295+oc013@users.noreply.github.com> Co-Authored-By: Aarni Koskela --- environment.yaml | 53 ++++++++++++++++++++++++------------------------ setup.py | 2 +- 2 files changed, 28 insertions(+), 27 deletions(-) diff --git a/environment.yaml b/environment.yaml index 06a4bcea2..ebf74757c 100644 --- a/environment.yaml +++ b/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 . \ No newline at end of file + - streamlit>=1.12.2 + - test-tube>=0.7.5 + - torch-fidelity==0.3.0 + - torchmetrics==0.6.0 + - transformers==4.19.2 \ No newline at end of file diff --git a/setup.py b/setup.py index a24d54167..0e768e157 100644 --- a/setup.py +++ b/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(),