a Ad1@s`ddlmZddlZddlZddlmZmZmZmZm Z e Z ddZ d ddZ d d d ZdS) wrapsN)sharedprogresserrorsdevices fifo_lockcsfdd}|S)Ncs6t|i|}Wdn1s(0Y|S)N) queue_lock)argskwargsresfunc:/var/www/html/stable-diffusion-webui/modules/call_queue.pyf s,zwrap_queued_call..fr)rrrr rwrap_queued_call s rcs"tfdd}t||ddS)Nc s|rDt|dtkrD|ddrD|ddrD|d}t|nd}thtjj |dt |z(|i|}t ||Wt |n t |0tj Wdn1s0Y|S)Nrztask())job)typestr startswithendswithrZadd_task_to_queuer rstatebegin start_taskZrecord_resultsZ finish_taskend)r r id_taskr r rrrs0  (zwrap_gradio_gpu_call..fT) extra_outputs add_stats)rwrap_gradio_call)rrrrr rwrap_gradio_gpu_callsr!Fcs t|dfdd }|S)N)extra_outputs_arrayc stjjdkotjj o}|r(tjt}zt|i|}Wnt y}zd}d}d|d|d|} t | |kr| d|dt | d7} t j |d | d d d tj _dtj _|durdd g}t|jd |} |dt| dg}WYd}~n d}~00tdtj _dtj _dtj _sFt|St|} t| d} | d} | dd}| dkr| d|}|r\ddtjD}|d}|d}|d}|d}|t|dd}d}d}d}d |d!|d"d#d$}d |d%|d"d#d$}d |d&|d"dd|d"d'd(|dd) }d*|d+|d+|d,}nd }|d-d.|d/|d7<t|S)0NrizError completing requestz Arguments:  z (Argument list truncated at /z characters) T)exc_infoz: z
z
F<z.1fz sec.z min. cSsi|]\}}||d qS)ir).0kvrrr ^z/wrap_gradio_call..f.. active_peak reserved_peak system_peaktotaldzRActive: peak amount of video memory used during generation (excluding cached data)zEReserved: total amout of video memory allocated by the Torch library z[System: peak amout of video memory allocated by all running programs, out of total capacityz A: iz.2fz GBz&'>R: z('>Sys: gz GB (z%)z

z, z

zT

Time taken: z

)roptsmemmon_poll_ratemem_mondisabledmonitortime perf_counterlist Exceptionlenrreportrr job_countr__name__htmlescapertorch_gcskipped interruptedtupleintstopitemsmax)r"r r Z run_memmontr eZmax_debug_str_lenmessagearg_str error_messageelapsedZ elapsed_m elapsed_sZ elapsed_textZ mem_statsr.r/Zsys_peakZ sys_totalZsys_pctZtoltip_aZtoltip_rZ toltip_sysZtext_aZtext_rZtext_sysZ vram_htmlrrrrr1s`  .    ,zwrap_gradio_call..fr)rrrrrrTrr 0sDr )N)NF) functoolsrrCr;modulesrrrrrZFIFOLockr rr!r rrrrs