return wrap_gradio_gpu_call to webui.py for extensions

1 files changed, 1 insertions, 1 deletions

diff --git a/webui.py b/webui.py
index 2eecfaa0..293a16cc 100644
--- a/webui.py
+++ b/webui.py
@@ -36,7 +36,7 @@ startup_timer.record("import ldm")
 

 from modules import extra_networks, ui_extra_networks_checkpoints

 from modules import extra_networks_hypernet, ui_extra_networks_hypernets, ui_extra_networks_textual_inversion

-from modules.call_queue import wrap_queued_call, queue_lock

+from modules.call_queue import wrap_gradio_gpu_call, wrap_queued_call, queue_lock  # noqa: F401

 

 # Truncate version number of nightly/local build of PyTorch to not cause exceptions with CodeFormer or Safetensors

 if ".dev" in torch.__version__ or "+git" in torch.__version__: