diff --git a/environment-cpuonly.yaml b/environment-cpuonly.yaml index f02e565..4805036 100644 --- a/environment-cpuonly.yaml +++ b/environment-cpuonly.yaml @@ -1,4 +1,4 @@ -name: ldm +name: sdco channels: - pytorch - defaults diff --git a/environment.yaml b/environment.yaml deleted file mode 100644 index 025ced8..0000000 --- a/environment.yaml +++ /dev/null @@ -1,31 +0,0 @@ -name: ldm -channels: - - pytorch - - defaults -dependencies: - - python=3.8.5 - - pip=20.3 - - cudatoolkit=11.3 - - pytorch=1.11.0 - - torchvision=0.12.0 - - numpy=1.19.2 - - pip: - - albumentations==0.4.3 - - diffusers - - opencv-python==4.1.2.30 - - pudb==2019.2 - - invisible-watermark - - imageio==2.9.0 - - imageio-ffmpeg==0.4.2 - - pytorch-lightning==1.4.2 - - omegaconf==2.1.1 - - test-tube>=0.7.5 - - streamlit>=0.73.1 - - einops==0.3.0 - - torch-fidelity==0.3.0 - - transformers==4.19.2 - - torchmetrics==0.6.0 - - kornia==0.6 - - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers - - -e git+https://github.com/openai/CLIP.git@main#egg=clip - - -e . diff --git a/install_sdco.sh b/install_sdco.sh new file mode 100644 index 0000000..a645669 --- /dev/null +++ b/install_sdco.sh @@ -0,0 +1,20 @@ +#!/bin/bash + +echo This Script is now installing a Stable Diffusion cpu only variant. + +mkdir -p models/ldm/stable-diffusion-v1/ +mkdir -p outputs/extras-samples +mkdir -p outputs/img2img-samples/samples +mkdir -p outputs/txt2img-samples/samples + + +conda install pytorch torchvision torchaudio cpuonly -c pytorch +conda env create -f environment-cpuonly.yaml +conda activate sdco + +pip install pynvml gradio keras-unet fairseq basicsr facexlib +pip install -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers +pip install -e git+https://github.com/openai/CLIP.git@main#egg=clip +pip install -e git+https://github.com/TencentARC/GFPGAN#egg=GFPGAN +pip install -e git+https://github.com/xinntao/Real-ESRGAN#egg=realesrgan +pip install -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion diff --git a/run_sdco.sh b/run_sdco.sh new file mode 100644 index 0000000..7134126 --- /dev/null +++ b/run_sdco.sh @@ -0,0 +1,4 @@ +#!/bin/bash +cd Home/$USER/stable-diffusion-cpuonly-main/ +conda activate sdco +python ./webui.py --no-half diff --git a/uninstall_sdco.sh b/uninstall_sdco.sh new file mode 100644 index 0000000..cf80d2b --- /dev/null +++ b/uninstall_sdco.sh @@ -0,0 +1,4 @@ +#!/bin/bash + +conda env remove sdco +