cpu only sd

app.py

@@ -41,6 +41,7 @@ def generate_drawing(text):
     args = clone_args()
     args.seed = random.randint(0, 2**32 - 1)
     args.prompt = text
+    print("Generating image... with text: " + str(text))
     results = sd.generate(args, return_c=True)
     c, image = results[0], results[1]
     size = c.size()

sd.py

@@ -22,7 +22,7 @@ from omegaconf import OmegaConf
 from huggingface_hub import hf_hub_download
 
 # 1. Download stable diffusion repository and set path
-os.system("git clone https://github.com/kael558/stable-diffusion")
+os.system("git clone https://github.com/darkhemic/stable-diffusion-cpuonly")
 os.system("git clone https://github.com/shariqfarooq123/AdaBins.git")
 os.system("git clone https://github.com/isl-org/MiDaS.git")
 os.system("git clone https://github.com/MSFTserver/pytorch3d-lite.git")
@@ -34,7 +34,7 @@ with open('k-diffusion/k_diffusion/__init__.py', 'w') as f:
 sys.path.extend([
     './taming-transformers',
     './clip',
-    'stable-diffusion/',
+    'stable-diffusion-cpuonly',
     'k-diffusion',
     'pytorch3d-lite',
     'AdaBins',