local iframeResizer

load iframeResizer from webui assets
ASSETS_COMMIT_HASH needs to be updated accordingly after https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets/pull/2 is merged

modules/initialize.py

@@ -14,7 +14,6 @@ def imports():
 
     import torch  # noqa: F401
     startup_timer.record("import torch")
-    from modules import patch_hf_hub_download  # noqa: F401
     import pytorch_lightning  # noqa: F401
     startup_timer.record("import torch")
     warnings.filterwarnings(action="ignore", category=DeprecationWarning, module="pytorch_lightning")

modules/patch_hf_hub_download.py

@@ -1,41 +0,0 @@
-from modules.patches import patch
-from modules.errors import report
-from inspect import signature
-from functools import wraps
-
-try:
-    from huggingface_hub.utils import LocalEntryNotFoundError
-    from huggingface_hub import file_download
-
-    def try_local_files_only(func):
-        if (param := signature(func).parameters.get('local_files_only', None)) and not param.kind == param.KEYWORD_ONLY:
-            raise ValueError(f'{func.__name__} does not have keyword-only parameter "local_files_only"')
-
-        @wraps(func)
-        def wrapper(*args, **kwargs):
-            try:
-                from modules.shared import opts
-                try_offline_mode = not kwargs.get('local_files_only') and opts.hd_dl_local_first
-            except Exception:
-                report('Error in try_local_files_only - skip try_local_files_only', exc_info=True)
-                try_offline_mode = False
-
-            if try_offline_mode:
-                try:
-                    return func(*args, **{**kwargs, 'local_files_only': True})
-                except LocalEntryNotFoundError:
-                    pass
-                except Exception:
-                    report('Unexpected exception in try_local_files_only - retry without patch', exc_info=True)
-
-            return func(*args, **kwargs)
-
-        return wrapper
-
-    try:
-        patch(__name__, file_download, 'hf_hub_download', try_local_files_only(file_download.hf_hub_download))
-    except RuntimeError:
-        pass  # already patched
-
-except Exception:
-    report('Error patching hf_hub_download', exc_info=True)

modules/shared_options.py

@@ -128,7 +128,6 @@ options_templates.update(options_section(('system', "System", "system"), {
     "disable_mmap_load_safetensors": OptionInfo(False, "Disable memmapping for loading .safetensors files.").info("fixes very slow loading speed in some cases"),
     "hide_ldm_prints": OptionInfo(True, "Prevent Stability-AI's ldm/sgm modules from printing noise to console."),
     "dump_stacks_on_signal": OptionInfo(False, "Print stack traces before exiting the program with ctrl+c."),
-    "hd_dl_local_first": OptionInfo(False, "Prevent connecting to huggingface for assets if cache is available").info('this will also prevent assets from being updated'),
 }))
 
 options_templates.update(options_section(('profiler', "Profiler", "system"), {

modules/ui_gradio_extensions.py

@@ -1,4 +1,5 @@
 import os
+import re
 import gradio as gr
 
 from modules import localization, shared, scripts, util
@@ -49,12 +50,21 @@ def css_html():
     return head
 
 
+re_preconnect = re.compile(rb'<link\s+rel="preconnect"\s+href="([^"]+)"(?:\s+[^>]*)?/?>')
+
+
 def reload_javascript():
     js = javascript_html()
     css = css_html()
 
     def template_response(*args, **kwargs):
         res = shared.GradioTemplateResponseOriginal(*args, **kwargs)
+
+        # remove preconnects
+        res.body = re_preconnect.sub(b'', res.body)
+        # replace iframeResizer with local version
+        res.body = res.body.replace(b'src="https://cdnjs.cloudflare.com/ajax/libs/iframe-resizer/4.3.6/iframeResizer.contentWindow.min.js"', b'src="webui-assets/js/iframe-resizer/4.3.6/iframeResizer.contentWindow.min.js"')
+
         res.body = res.body.replace(b'</head>', f'{js}<meta name="referrer" content="no-referrer"/></head>'.encode("utf8"))
         res.body = res.body.replace(b'</body>', f'{css}</body>'.encode("utf8"))
         res.init_headers()