обработка ошибки download

scripts/mainrunpodA1111.py

@@ -525,14 +525,16 @@ def InstallWildCards():
 
 def CNet(ControlNet_Model, ControlNet_XL_Model):
     def download(url, model_dir):
-
-        filename = os.path.basename(urlparse(url).path)
-        pth = os.path.abspath(os.path.join(model_dir, filename))
-        if not os.path.exists(pth):
-            print("Downloading: " + os.path.basename(url))
-            download_url_to_file(url, pth, hash_prefix=None, progress=True)
-        else:
-            print(f"[1;32mThe model {filename} already exists[0m")
+        try:
+            filename = os.path.basename(urlparse(url).path)
+            pth = os.path.abspath(os.path.join(model_dir, filename))
+            if not os.path.exists(pth):
+                print("Downloading: " + os.path.basename(url))
+                download_url_to_file(url, pth, hash_prefix=None, progress=True)
+            else:
+                print(f"[1;32mThe model {filename} already exists[0m")
+        except Exception as e:
+            print(f"Ошибка при скачивании {url}: {e}")
 
     wrngv1 = False
     os.chdir("/workspace/sd/stable-diffusion-webui/extensions")