From f06888d39df633edbb0f17295cffdbd26964e36a Mon Sep 17 00:00:00 2001
From: Christopher Williams <darkhemic@gmail.com>
Date: Sat, 1 Oct 2022 09:36:42 -0400
Subject: [PATCH] easy to use scripts started

---
 environment-cpuonly.yaml |  2 +-
 environment.yaml         | 31 -------------------------------
 install_sdco.sh          | 20 ++++++++++++++++++++
 run_sdco.sh              |  4 ++++
 uninstall_sdco.sh        |  4 ++++
 5 files changed, 29 insertions(+), 32 deletions(-)
 delete mode 100644 environment.yaml
 create mode 100644 install_sdco.sh
 create mode 100644 run_sdco.sh
 create mode 100644 uninstall_sdco.sh

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
+