From f367b07282a3cf4fb6a2781b1e02ca0299fa7202 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Fri, 2 Aug 2024 11:46:36 -0700
Subject: [PATCH] fix gradio url timeout

#909
---
 modules_forge/patch_basic.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/modules_forge/patch_basic.py b/modules_forge/patch_basic.py
index bb530325..87048a20 100644
--- a/modules_forge/patch_basic.py
+++ b/modules_forge/patch_basic.py
@@ -1,8 +1,26 @@
 import torch
 import os
+import time
+import httpx
+import warnings
+import gradio.networking
 import safetensors.torch
 
 
+def gradio_url_ok_fix(url: str) -> bool:
+    try:
+        for _ in range(5):
+            with warnings.catch_warnings():
+                warnings.filterwarnings("ignore")
+                r = httpx.head(url, timeout=999, verify=False)
+            if r.status_code in (200, 401, 302):
+                return True
+            time.sleep(0.500)
+    except (ConnectionError, httpx.ConnectError):
+        return False
+    return False
+
+
 def build_loaded(module, loader_name):
     original_loader_name = loader_name + '_origin'
 
@@ -38,6 +56,7 @@ def build_loaded(module, loader_name):
 
 
 def patch_all_basics():
+    gradio.networking.url_ok = gradio_url_ok_fix
     build_loaded(safetensors.torch, 'load_file')
     build_loaded(torch, 'load')
     return