Commit fa9d2ac2 authored by Jared Deckard's avatar Jared Deckard
Browse files

Fix gradio special args in the call queue

parent baf6946e
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
from functools import wraps
import html
import sys
import threading
@@ -20,6 +21,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
@@ -47,6 +49,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: