o پi9@sdZddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddlm Z ddl mZddlmZmZddlmmZejZejZdZd Zd Zd Zd Zed ZdZddeedidddeddiddgdddidgdddddZGdddejZ Gdddej!Z"e d e d!e#d"dfd#d$Z$e d e d!e#d"dfd%d&Z%d'd(Z&d)d*Z'd+e(d,e(d-e d!e)d.ed/e*d0e*d1ed"dfd2d3Z+Gd4d5d5e Z,d6e#d"e,fd7d8Z-e-e.Z/dZd9d:Z0dZd;e#dZ1d?d@Z2d[dBe#fdCdDZ3e ddEd"e(fdFdGZ4ej5fdHe6e#d,e(fdIdJZ7dKdLZ8ejdMdNZ9GdOdPdPZ:e  d\d ej dQe#dRe(dBdSe(dBfdTdUZ;d ej dVe(dWe`sz3SortedHelpFormatter.add_arguments..)key)sortedr add_arguments)r#actionsr&r(r)r<_sz!SortedHelpFormatter.add_arguments)r*r+r,r-r<r4r(r(r&r)r5\sr5loggermsgrcC|j|dddSN) stacklevel)infor>r?r(r(r)_print_info_oncedrFcCr@rA)warningrEr(r(r)_print_warning_oncejrGrIc C:z ttjd}W|dkSttfyd}Y|dkSw)NRANKrintosenvironKeyError ValueErrorrankr(r(r)get_is_main_processprTc CrJ)N LOCAL_RANKrrLrRr(r(r)get_is_local_main_processxrUrWserver_log_levelr logger_selfargsmain_process_onlylocal_main_process_onlykwargsc Ost}t} | r | p|o|p|o| p|tjk} | r?d|vr.|j||g|Ri|dS|j||g|Rddi|dSdS)zJHelper function to log a message if the process rank matches the criteria.rCN)rTrWr.rlog) rXrrYr?r[r\rZr]is_main_processis_local_main_process should_logr(r(r)_log_process_awares  "rcc @seZdZdZdeddfddZdeddfddZd d d ded ed e d e deddf ddZ d d d ded ed e d e deddf ddZ dd d ded ed e d e deddf ddZ dd d ded ed e d e deddf ddZ dS)_SGLDiffusionLoggera  Note: This class is just to provide type information. We actually patch the methods directly on the :class:`logging.Logger` instance to avoid conflicting with other libraries such as `intel_extension_for_pytorch.utils._logger`. r?rNcCt||dS)zk As :meth:`info`, but subsequent calls with the same message are silently dropped. N)rFr#r?r(r(r) info_oncez_SGLDiffusionLogger.info_oncecCre)zn As :meth:`warning`, but subsequent calls with the same message are silently dropped. N)rIrfr(r(r) warning_oncerhz _SGLDiffusionLogger.warning_onceTr[r\rZr[r\r]cOdSr6r(r#r?r[r\rZr]r(r(r)rDz_SGLDiffusionLogger.infocOrkr6r(rlr(r(r)debugrmz_SGLDiffusionLogger.debugFcOrkr6r(rlr(r(r)rHrmz_SGLDiffusionLogger.warningcOrkr6r(rlr(r(r)errorrmz_SGLDiffusionLogger.error)r*r+r,r-r3rgriobjectrboolrDrnrHror(r(r(r)rds       rdnamecst|}|t|dtt|t|dtt|dtdtdtffdd }t|dt|tj d d |t|d t|tj d d |t|d t|tj d d |t|d t|tj d d |t t|S)zThe main purpose of this function is to ensure that loggers are retrieved in such a way that we can be sure the root sgl_diffusion logger has already been configured.rgrirmain_process_only_defaultlocal_main_process_only_defaultcs8||ddtdtdtdtdtdtddffd d }|S) Nrjr#r?rZr[r\r]rcs&t||g|R||d|dS)Nrj)rcrl)rrXr(r)_methods z._create_patched_method.._method)rrprrq)rrsrtrurX)rr)_create_patched_methods$ z+init_logger.._create_patched_methodrDTrnrHFro)r. getLoggergetEffectiveLevelsetattrrrFrIrMrqINFOrrr/rrd)rrr>rwr(rvr) init_loggers@  r|cCs.|dvr|jj}|j}|jj}||sdSzq|j}|dur*|jj} |j} |jj} nd} d} d} t|dI} tj d} |dkr]| | d|d|d |d | d| d | d n| | d |d|d |d | d| d | d Wdn1swYWn t yYnwt t ||S)N)callrraz%Y-%m-%d %H:%M:%S.%fr}z Call to z in :z from  z Return from z to )f_code co_filenamef_linenoco_name startswithf_backopendatetimenowstrftimewrite NameErrorr _trace_calls)log_pathroot_dirframeeventargfilenamelineno func_name last_frame last_filename last_linenolast_func_nameftsr(r(r)rsn     r log_file_pathrcCsHtdtd||durtjtjt}tt t ||dS)a Enable tracing of every function call in code under `root_dir`. This is useful for debugging hangs or crashes. `log_file_path` is the path to the log file. `root_dir` is the root directory of the code to trace. If None, it is the sgl_diffusion root directory. Note that this call is thread-level, any threads calling this function will have the trace enabled. Other threads will not be affected. zSGLANG_DIFFUSION_TRACE_FUNCTION is enabled. It will record every function executed by Python. This will slow down the code. It is suggested to be used for debugging hang or crashes only.zTrace frame log is saved to %sN) r>rHrDrNpathdirname__file__syssettracerr)rrr(r(r)enable_trace_function_callGs  rcCsPddlm}d|ddd<d|ddd<d |dd d<d|dd d<dS) NrLOGGING_CONFIGz)[%(asctime)s] %(levelprefix)s %(message)srdefaultfmtz%Y-%m-%d %H:%M:%Sr zR[%(asctime)s] %(levelprefix)s %(client_addr)s - "%(request_line)s" %(status_code)saccess)uvicorn.configrrr(r(r)set_uvicorn_logging_configs^s   rr~prefixcCsnd|d}d}t||d}ttj}||t}|j| || t t|j tdS)Nz [%(asctime)sz ] %(message)sr )r )rr. StreamHandlerrstdout setFormatterrxrclear addHandlersetLevelgetattr log_levelupperr) server_argsr log_formatr rhandlerrr(r(r)configure_loggerks       r)maxsizecCst}|jSr6)r.rxr)rr(r(r) get_log_level{srloggers_to_suppresscCs0i}|D]}t|}|j||<||q|Sr6)r.rxrr)rroriginal_levels logger_namer>r(r(r)suppress_loggerss    rcCs(gd}|D] }t|tjqdS)N)imageioimageio_ffmpegPIL PIL_Imagezpython_multipart.multipartfilelockurllib3)r.rxrr/) target_namesrrr(r(r)globally_suppress_loggerss rc cstj}t|}ttjtj}z&tjt ||dVWtjt ||t |t |dStjt ||t |t |w)a; Suppress stdout from C libraries at the file descriptor level. Only suppresses stdout, not stderr, to preserve error messages. Example: with suppress_stdout(): # C library calls that would normally print to stdout torch.distributed.new_group(ranks, backend="gloo") N) rrfilenorNduprdevnullO_WRONLYflushdup2close) stdout_fd stdout_dup devnull_fdr(r(r)suppress_stdouts          rc@seZdZddZdS)GenerationTimercCsd|_d|_d|_dS)Ng) start_timeend_timeduration)r#r(r(r)__init__s zGenerationTimer.__init__N)r*r+r,rr(r(r(r)rs rprompt request_idxtotal_requestsc cs|dur|dur|d|||ddt}t|_z |Vt|_|j|j|_|dtdtd|jWdSt yb}z|durS|j d||dd|j d |ddd}~ww) NzProcessing prompt %d/%d: %sdz%Pixel data generated successfully in %.2f secondsz+Failed to generate output for prompt %d: %sT)exc_infoz&Failed to generate output for prompt: ) rDrtime perf_counterrrrGREENr" Exceptionro)r>rrrtimerer(r(r)log_generation_timersD     r num_outputs total_timecCs |dtdtd||dS)Nz4Completed batch processing. Generated %d outputs in rr)rDrr")r>rrr(r(r)log_batch_completions rr6)r~)NN)>r-argparse contextlibrr.rNrrr functoolsrrrtypesrtypingrrsglang.multimodal_gen.envsmultimodal_genenvsSGLANG_DIFFUSION_LOGGING_LEVELSGLANG_DIFFUSION_LOGGING_PREFIXCYANr0rr1r"_FORMAT _DATE_FORMATDEFAULT_LOGGING_CONFIG Formatterr HelpFormatterr5r3rFrIrTrWrMrprqrcrdr|r*r>rrrrrrlistrrrrrfloatrr(r(r(r)s      <? +    (