aboutsummaryrefslogtreecommitdiffstats
path: root/modules/call_queue.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-23 12:48:09 +0000
committerAUTOMATIC1111 <16777216c@gmail.com>2023-08-23 12:48:09 +0000
commitc9c8485bc1e8720aba70f029d25cba1c4abf2b5c (patch)
tree572e29867d4df32cf33df284cf96e8611b78045c /modules/call_queue.py
parent68f336bd994bed5442ad95bad6b6ad5564a5409a (diff)
parent31f2be3dcedf85c036c5f784c640208d122b62ed (diff)
downloadstable-diffusion-webui-gfx803-c9c8485bc1e8720aba70f029d25cba1c4abf2b5c.tar.gz
stable-diffusion-webui-gfx803-c9c8485bc1e8720aba70f029d25cba1c4abf2b5c.tar.bz2
stable-diffusion-webui-gfx803-c9c8485bc1e8720aba70f029d25cba1c4abf2b5c.zip
Merge branch 'release_candidate'
Diffstat (limited to 'modules/call_queue.py')
-rw-r--r--modules/call_queue.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/call_queue.py b/modules/call_queue.py
index 61aa240f..f2eb17d6 100644
--- a/modules/call_queue.py
+++ b/modules/call_queue.py
@@ -3,7 +3,7 @@ import html
import threading
import time
-from modules import shared, progress, errors
+from modules import shared, progress, errors, devices
queue_lock = threading.Lock()
@@ -75,6 +75,8 @@ def wrap_gradio_call(func, extra_outputs=None, add_stats=False):
error_message = f'{type(e).__name__}: {e}'
res = extra_outputs_array + [f"<div class='error'>{html.escape(error_message)}</div>"]
+ devices.torch_gc()
+
shared.state.skipped = False
shared.state.interrupted = False
shared.state.job_count = 0