easy to use scripts started

environment-cpuonly.yaml

@@ -1,4 +1,4 @@
-name: ldm
+name: sdco
 channels:
   - pytorch
   - defaults

environment.yaml

@@ -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 .

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

run_sdco.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+cd Home/$USER/stable-diffusion-cpuonly-main/
+conda activate sdco
+python ./webui.py --no-half

uninstall_sdco.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+
+conda env remove sdco
+