o ۷i'@sDUdZddlZddlZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl mZddlmZeZdaejdBed<ejej ej ejejdZej Zd adaGd d d ejZd ejd dfddZd efddZ d e!fddZ"d ejfddZ#dIddZ$dIddZ%d e&e!effddZ'dJde!dBd ejfddZ(d efdd Z)d!ed dfd"d#Z*dId$d%Z+dId&d'Z,dId(d)Z-dId*d+Z.dId,d-Z/dId.d/Z0d0ejd dfd1d2Z1d0ejd dfd3d4Z2dId5d6Z3dId7d8Z4dId9d:Z5dId;d<Z6dId=d>Z7e7ej_7Gd?d@d@Z8GdAdBdBZ9e9Z d e:fdCdDZ;dIdEdFZBsz+_ensure_rank_zero_filter..)_rank_zero_filterranyfilters addFilter)rrrr_ensure_rank_zero_filter<s r)c CsHtdd}|r"|tvrt|Std|ddttS)z If DIFFUSERS_VERBOSITY env var is set to one of the valid choices return that as the new default level. If it is not - fall back to `_default_log_level` DIFFUSERS_VERBOSITYNz#Unknown option DIFFUSERS_VERBOSITY=z, has to be one of: z, ) osgetenv log_levelsr getLoggerrjoinkeys_default_log_level) env_level_strrrr_get_default_logging_levelFs r3cCstddS)N.r)rsplitrrrr_get_library_nameVsr6cCs ttSr)rr.r6rrrr_get_library_root_loggerZs r7cCst5tr WddStatjrtjjt_t}|t| t d|_ t |WddS1s:wYdS)NF) _lockr r StreamHandlersysstderrflushr7 addHandlersetLevelr3 propagater)library_root_loggerrrr_configure_library_root_logger^s    "rBcCs`t$ts WddSt}|t|tjdaWddS1s)wYdSr)r8r r7 removeHandlerr>rrr@rrr_reset_library_root_loggerrs  "rDcCstSr)r-rrrrget_log_levels_dictrEnamecCs*|durt}tt|}t||S)z Return a logger with the specified name. This function is not supposed to be directly accessed unless you are writing a custom diffusers module. N)r6rBrr.r))rGrrrr get_loggers  rHcCsttS)u Return the current level for the 🤗 Diffusers' root logger as an `int`. Returns: `int`: Logging level integers which can be one of: - `50`: `diffusers.logging.CRITICAL` or `diffusers.logging.FATAL` - `40`: `diffusers.logging.ERROR` - `30`: `diffusers.logging.WARNING` or `diffusers.logging.WARN` - `20`: `diffusers.logging.INFO` - `10`: `diffusers.logging.DEBUG` )rBr7getEffectiveLevelrrrr get_verbositys rJ verbositycCstt|dS)u Set the verbosity level for the 🤗 Diffusers' root logger. Args: verbosity (`int`): Logging level which can be one of: - `diffusers.logging.CRITICAL` or `diffusers.logging.FATAL` - `diffusers.logging.ERROR` - `diffusers.logging.WARNING` or `diffusers.logging.WARN` - `diffusers.logging.INFO` - `diffusers.logging.DEBUG` N)rBr7r>)rKrrr set_verbositysrLcCttS)z&Set the verbosity to the `INFO` level.)rLrrrrrset_verbosity_inforNcCrM)z)Set the verbosity to the `WARNING` level.)rLr rrrrset_verbosity_warningrOrPcCrM)z'Set the verbosity to the `DEBUG` level.)rLrrrrrset_verbosity_debugrOrQcCrM)z'Set the verbosity to the `ERROR` level.)rLrrrrrset_verbosity_errorrOrRcC"ttdus JttdS)u?Disable the default handler of the 🤗 Diffusers' root logger.N)rBr r7rCrrrrdisable_default_handler rTcCrS)u>Enable the default handler of the 🤗 Diffusers' root logger.N)rBr r7r=rrrrenable_default_handlerrUrVhandlercCs"t|dus Jt|dS)z9adds a handler to the HuggingFace Diffusers' root logger.N)rBr7r=rWrrr add_handlerrUrYcCs.t|dur |tjvsJt|dS)zBremoves given handler from the HuggingFace Diffusers' root logger.N)rBr7handlersrCrXrrrremove_handlersr[cCtdt_dS)zk Disable propagation of the library log outputs. Note that log propagation is disabled by default. FNrBr7r?rrrrdisable_propagations r^cCr\)z Enable propagation of the library log outputs. Please disable the HuggingFace Diffusers' default handler to prevent double logging if the root logger has been configured. TNr]rrrrenable_propagations r_cCs*tj}|D] }td}||qdS)u Enable explicit formatting for every 🤗 Diffusers' logger. The explicit formatter is as follows: ``` [LEVELNAME|FILENAME|LINE NUMBER] TIME >> MESSAGE ``` All handlers currently bound to the root logger are affected by this method. zB[%(levelname)s|%(filename)s:%(lineno)s] %(asctime)s >> %(message)sN)r7rZr Formatter setFormatter)rZrW formatterrrrenable_explicit_formats   rccCs tj}|D]}|dqdS)u Resets the formatting for 🤗 Diffusers' loggers. All handlers currently bound to the root logger are affected by this method. N)r7rZra)rZrWrrr reset_formats rdcOs(tdd}|r dS|j|i|dS)z This method is identical to `logger.warning()`, but if env var DIFFUSERS_NO_ADVISORY_WARNINGS=1 is set, this warning will not be printed DIFFUSERS_NO_ADVISORY_WARNINGSFN)r+r,r)rargskwargsno_advisory_warningsrrrwarning_advice s ric@s8eZdZdZddZddZddZdd Zd d Zd S) EmptyTqdmz%Dummy tqdm which doesn't do anything.cOs|r |d|_dSd|_dS)Nr) _iteratorrrfrgrrr__init__1szEmptyTqdm.__init__cCs t|jSr)iterrkrrrr__iter__4s zEmptyTqdm.__iter__cCs dd}|S)zReturn empty function.c_dSrr)rfrgrrrempty_fn:rFz'EmptyTqdm.__getattr__..empty_fnr)r_rrrrr __getattr__7szEmptyTqdm.__getattr__cCs|Srrrorrr __enter__?rFzEmptyTqdm.__enter__cCrqrr)rtype_value tracebackrrr__exit__BrFzEmptyTqdm.__exit__N) rrr__doc__rmrprtruryrrrrrj.s rjc@s$eZdZddZddZddZdS) _tqdm_clscOs"tr tj|i|St|i|Sr) _tqdm_activetqdm_libtqdmrjrlrrr__call__Gsz_tqdm_cls.__call__cOs d|_trtjj|i|SdSr)r8r|r}r~set_lockrlrrrrMsz_tqdm_cls.set_lockcCstrtjSdSr)r|r}r~get_lockrorrrrRs z_tqdm_cls.get_lockN)rrrrrrrrrrr{Fs r{cCrM)zCReturn a boolean indicating whether tqdm progress bars are enabled.)boolr|rrrris_progress_bar_enabledZrcCdadS)zEnable tqdm progress bar.TNr|rrrrenable_progress_bar`rrcCr)zDisable tqdm progress bar.FNrrrrrdisable_progress_barfrr)r Nr)>rzrr+r: threadingrrrrrrrr r~r r}distributed_utilsr Lockr8r Handler__annotations__r-r1r|r%FilterrLoggerr)intr3strr6r7rBrDdictrErHrJrLrNrPrQrRrTrVrYr[r^r_rcrdrirjr{rrrrrrrrsd(