diff --git a/install_sdco.sh b/install_sdco.sh
index 2c60041..021d124 100755
--- a/install_sdco.sh
+++ b/install_sdco.sh
@@ -11,6 +11,8 @@ mkdir -p outputs/txt2img-samples/samples
 #conda env create -f environment-cpuonly.yaml
 #conda activate sdco
 
+conda install pytorch torchvision torchaudio cpuonly -c pytorch
+
 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
@@ -18,6 +20,10 @@ 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
 
+pip install chardet
+pip install --force-reinstall charset-normalizer==3.1.0
+pip install git+https://github.com/crowsonkb/k-diffusion/
+
 #cp sd-v1-4.ckpt models/ldm/stable-diffusion-v1/model.ckpt
 #rm sd-v1-4.ckpt
 #cp GFPGANv1.3.pth src/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth