aboutsummaryrefslogtreecommitdiff
path: root/modules/call_queue.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-06-27 09:06:03 +0300
committerGitHub <noreply@github.com>2023-06-27 09:06:03 +0300
commitd35e24611172a49f7b3637c601dc0fb12c9d0326 (patch)
treee7915af4d068912cd8509f1638e05460445a5eea /modules/call_queue.py
parentbedcd2f377a38ef4da58c11dbe222d32b954be2f (diff)
parent4147fd6b2f905f76c6bc20c3d9de2ea0842fa853 (diff)
Merge pull request #11227 from deckar01/10141-gradio-user-exif
Add Gradio User to Metadata
Diffstat (limited to 'modules/call_queue.py')
-rw-r--r--modules/call_queue.py3
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: