Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -240,7 +240,54 @@ def mdl(Original_Model_Version, Path_to_MODEL, MODEL_LINK):
         model="/workspace/sd/stable-diffusion-webui/models/Stable-diffusion"
 
     return model
+    
+def modeldwn(model_LINK):
+
+    if model_LINK == "":
+        print("[1;33mNothing to do")
+    else:
+        os.makedirs(
+            "/workspace/sd/stable-diffusion-webui/models/Stable-diffusion",
+            exist_ok=True,
+        )
+
+        src = getsrc(model_LINK)
+
+        if src == "civitai":
+            modelname = get_name(model_LINK, False)
+            loramodel = f"/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/{modelname}"
+            if not os.path.exists(loramodel):
+                dwn(model_LINK, loramodel, "Downloading the LoRA model")
+                clear_output()
+            else:
+                print("[1;33mModel already exists")
+        elif src == "gdrive":
+            modelname = get_true_name(model_LINK)
+            loramodel = f"/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/{modelname}"
+            if not os.path.exists(loramodel):
+                gdown.download(
+                    url=model_LINK.replace("/file/d/", "/uc?id=").replace("/view", ""),
+                    output=loramodel,
+                    quiet=False,
+                )
+                clear_output()
+            else:
+                print("[1;33mModel already exists")
+        else:
+            modelname = os.path.basename(model_LINK)
+            loramodel = f"/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/{modelname}"
+            if not os.path.exists(loramodel):
+                gdown.download(
+                    url=model_LINK, output=loramodel, quiet=False, fuzzy=True
+                )
+                clear_output()
+            else:
+                print("[1;33mModel already exists")
 
+        if os.path.exists(loramodel):
+            print("[1;32mCheckpoints downloaded")
+        else:
+            print("[1;31mWrong link, check that the link is valid")
 
 def loradwn(LoRA_LINK):