diff options
| author | Beinsezii <39478211+Beinsezii@users.noreply.github.com> | 2023-06-27 22:29:47 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-06-27 22:29:47 +0000 |
| commit | 9d8af4bd6aaf09b8a94dc10dd5e99c82e23dec38 (patch) | |
| tree | d86d470db2289a31e589a7bd777a38784605313c /modules/call_queue.py | |
| parent | 1d7c51fb9f757b5dcdc506f8fc003e6047151567 (diff) | |
| parent | fab73f2e7d388ca99cdb3d5de7f36c0b9a1a3b1c (diff) | |
| download | stable-diffusion-webui-gfx803-9d8af4bd6aaf09b8a94dc10dd5e99c82e23dec38.tar.gz stable-diffusion-webui-gfx803-9d8af4bd6aaf09b8a94dc10dd5e99c82e23dec38.tar.bz2 stable-diffusion-webui-gfx803-9d8af4bd6aaf09b8a94dc10dd5e99c82e23dec38.zip | |
Merge branch 'AUTOMATIC1111:dev' into dev
Diffstat (limited to 'modules/call_queue.py')
| -rw-r--r-- | modules/call_queue.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/call_queue.py b/modules/call_queue.py index 1b5e5273..69bf63d2 100644 --- a/modules/call_queue.py +++ b/modules/call_queue.py @@ -1,3 +1,4 @@ +from functools import wraps
import html
import threading
import time
@@ -18,6 +19,7 @@ def wrap_queued_call(func): def wrap_gradio_gpu_call(func, extra_outputs=None):
+ @wraps(func)
def f(*args, **kwargs):
# if the first argument is a string that says "task(...)", it is treated as a job id
@@ -45,6 +47,7 @@ def wrap_gradio_gpu_call(func, extra_outputs=None): def wrap_gradio_call(func, extra_outputs=None, add_stats=False):
+ @wraps(func)
def f(*args, extra_outputs_array=extra_outputs, **kwargs):
run_memmon = shared.opts.memmon_poll_rate > 0 and not shared.mem_mon.disabled and add_stats
if run_memmon:
|