Skip to content

S
Stable Diffusion Webui
Commit fa9d2ac2 authored by Jared Deckard's avatar Jared Deckard
Browse files
Options

Fix gradio special args in the call queue

parent baf6946e
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
modules/call_queue.py
View file @ fa9d2ac2
from functools import wraps
import html import html
import sys import sys
import threading import threading
...@@ -20,6 +21,7 @@ def wrap_queued_call(func): ...@@ -20,6 +21,7 @@ def wrap_queued_call(func):
def wrap_gradio_gpu_call(func, extra_outputs=None): def wrap_gradio_gpu_call(func, extra_outputs=None):
@wraps(func)
def f(*args, **kwargs): def f(*args, **kwargs):
# if the first argument is a string that says "task(...)", it is treated as a job id # if the first argument is a string that says "task(...)", it is treated as a job id
...@@ -47,6 +49,7 @@ def wrap_gradio_gpu_call(func, extra_outputs=None): ...@@ -47,6 +49,7 @@ def wrap_gradio_gpu_call(func, extra_outputs=None):
def wrap_gradio_call(func, extra_outputs=None, add_stats=False): def wrap_gradio_call(func, extra_outputs=None, add_stats=False):
@wraps(func)
def f(*args, extra_outputs_array=extra_outputs, **kwargs): 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 run_memmon = shared.opts.memmon_poll_rate > 0 and not shared.mem_mon.disabled and add_stats
if run_memmon: if run_memmon:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment