From baf51e46816a209407d19aa69718764d54b20d13 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Thu, 9 May 2024 02:24:06 +0200 Subject: [PATCH] sd upgraded to add controlnet --- lollms/services/sd/lollms_sd.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/lollms/services/sd/lollms_sd.py b/lollms/services/sd/lollms_sd.py index 82aa7e4..07aef16 100644 --- a/lollms/services/sd/lollms_sd.py +++ b/lollms/services/sd/lollms_sd.py @@ -80,6 +80,7 @@ def install_sd(lollms_app:LollmsApplication): else: print("Cloning repository...") subprocess.run(["git", "clone", "https://github.com/ParisNeo/stable-diffusion-webui.git", str(sd_folder)]) + subprocess.run(["git", "clone", "https://github.com/Mikubill/sd-webui-controlnet", str(sd_folder/"extensions/sd-webui-controlnet")]) subprocess.run(["git", "clone", "https://github.com/deforum-art/sd-webui-deforum.git", str(sd_folder/"extensions/sd-webui-deforum")]) subprocess.run(["git", "clone", "https://github.com/ParisNeo/SD-CN-Animation.git", str(sd_folder/"extensions/SD-CN-Animation")]) @@ -106,6 +107,10 @@ def upgrade_sd(lollms_app:LollmsApplication): except: subprocess.run(["git", "clone", "https://github.com/deforum-art/sd-webui-deforum.git", str(sd_folder/"extensions/sd-webui-deforum")]) + try: + subprocess.run(["git", "-C", str(sd_folder/"extensions/sd-webui-controlnet"), "pull"]) + except: + subprocess.run(["git", "clone", "https://github.com/Mikubill/sd-webui-controlnet", str(sd_folder/"extensions/sd-webui-controlnet")]) try: subprocess.run(["git", "-C", str(sd_folder/"extensions/SD-CN-Animation"), "pull"])