diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-04 05:05:21 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-04 05:05:21 +0000 |
| commit | 56c3f94ba30d76bf680db9bc765624b9a143d769 (patch) | |
| tree | 6924fc635fe13fe59911a285313308be2d9acdc7 /modules/call_queue.py | |
| parent | 952effa8b10dba2f2f7f2cf4191f987e3666e9f0 (diff) | |
| parent | 073c0ebba380acbd73be8262feba41212165ff84 (diff) | |
| download | stable-diffusion-webui-gfx803-56c3f94ba30d76bf680db9bc765624b9a143d769.tar.gz stable-diffusion-webui-gfx803-56c3f94ba30d76bf680db9bc765624b9a143d769.tar.bz2 stable-diffusion-webui-gfx803-56c3f94ba30d76bf680db9bc765624b9a143d769.zip | |
Merge branch 'dev' into dev
Diffstat (limited to 'modules/call_queue.py')
| -rw-r--r-- | modules/call_queue.py | 4 |
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
|