return wrap_gradio_gpu_call to webui.py for extensions

This commit is contained in:
AUTOMATIC 2023-05-11 10:09:29 +03:00
parent 3bb964d806
commit 87c3aa7389

View File

@ -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__: