(ti'PUdZddlZddlZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl mZddlmZej$Zdaej*dzed<ej ejejejej dZejZd adaGd d ej6Zd ej:d dfdZd efdZ d e!fdZ"d ej:fdZ#d/dZ$d/dZ%d e&e!effdZ'd0de!dzd ej:fdZ(d efdZ)ded dfdZ*d/dZ+d/dZ,d/dZ-d/dZ.d/dZ/d/dZ0d ej*d dfd!Z1d ej*d dfd"Z2d/d#Z3d/d$Z4d/d%Z5d/d&Z6d/d'Z7e7ej:_7Gd(d)Z8Gd*d+Z9e9Z d e:fd,Z;d/d-Zz+_ensure_rank_zero_filter..BsF~!z!_-~s)_rank_zero_filterranyfilters addFilter)r#s r_ensure_rank_zero_filterr/<s: +- Fv~~F F*+ Grc tjdd}|r[|tvr t|Stjj d|ddj tjtS)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_strs r_get_default_logging_levelr:Fsr II3T:M J &m, ,     ' '5m_DXY]YbYbcmcrcrctYuXvw  rc2tjddS)N.r)rsplitr"rr_get_library_namer>Vs >># q !!rc<tjtSr)rr5r>r"rr_get_library_root_loggerr@Zs   .0 11rc|t5tr dddytjatj r#tj j t_t}|jt|jtd|_ t|dddy#1swYyxYw)NF) _lockrr StreamHandlersysstderrflushr@ addHandlersetLevelr: propagater/library_root_loggers r_configure_library_root_loggerrL^s    #002 ::%(ZZ%5%5  "78&&'78$$%?%AB(-% !45 sB2BB22B;ct5ts dddyt}|jt|j t j dadddy#1swYyxYwr)rBrr@ removeHandlerrHrrrJs r_reset_library_root_loggerrOrsP   78))*:;$$W^^4 sA!AA!!A*ctSr)r4r"rrget_log_levels_dictrQs rnamecr| t}ttj|}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. )r>rLrr5r/)rRr#s r get_loggerrTs4 | ""$   t $FV$ MrcFttjS)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` )rLr@getEffectiveLevelr"rr get_verbosityrWs #$ # % 7 7 99r verbositycJttj|y)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)rLr@rH)rXs r set_verbosityrZs#$'' 2rc ttS)z&Set the verbosity to the `INFO` level.)rZrr"rrset_verbosity_infor\s  rc ttS)z)Set the verbosity to the `WARNING` level.)rZr r"rrset_verbosity_warningr^s  !!rc ttS)z'Set the verbosity to the `DEBUG` level.)rZrr"rrset_verbosity_debugr`  rc ttS)z'Set the verbosity to the `ERROR` level.)rZrr"rrset_verbosity_errorrcrarcbttJtjty)u?Disable the default handler of the 🤗 Diffusers' root logger.N)rLrr@rNr"rrdisable_default_handlerres(#$  '' ',,-=>rcbttJtjty)u>Enable the default handler of the 🤗 Diffusers' root logger.N)rLrr@rGr"rrenable_default_handlerrgs(#$  '' '))*:;rhandlercRt|Jtj|y)z9adds a handler to the HuggingFace Diffusers' root logger.N)rLr@rGrhs r add_handlerrks'#$   ))'2rc~t||tjvsJtj|y)zBremoves given handler from the HuggingFace Diffusers' root logger.N)rLr@handlersrNrjs rremove_handlerrns8#$  7.F.H.Q.Q#QQ Q,,W5rc6tdt_y)zk Disable propagation of the library log outputs. Note that log propagation is disabled by default. FNrLr@rIr"rrdisable_propagationrqs #$+0(rc6tdt_y)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. TNrpr"rrenable_propagationrss #$+/(rctj}|D](}tjd}|j |*y)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)r@rmr Formatter setFormatter)rmrh formatters renable_explicit_formatrxs;()22H%%&jk Y'rc\tj}|D]}|jdy)u Resets the formatting for 🤗 Diffusers' loggers. All handlers currently bound to the root logger are affected by this method. N)r@rmrv)rmrhs r reset_formatrzs* ()22HT"rcZtjdd}|ry|j|i|y)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)r2r3r)rargskwargsno_advisory_warningss rwarning_advicer s/ 99%EuMDLL$!&!rc.eZdZdZdZdZdZdZdZy) EmptyTqdmz%Dummy tqdm which doesn't do anything.c,|r |d|_yd|_y)Nr) _iteratorrr}r~s r__init__zEmptyTqdm.__init__1s$(adrc,t|jSr)iterrrs r__iter__zEmptyTqdm.__iter__4sDNN##rc d}|S)zReturn empty function.cyrr")r}r~s rempty_fnz'EmptyTqdm.__getattr__..empty_fn:s rr")r_rs r __getattr__zEmptyTqdm.__getattr__7s rc|Srr"rs r __enter__zEmptyTqdm.__enter__?s rcyrr")rtype_value tracebacks r__exit__zEmptyTqdm.__exit__BsrN) rr r!__doc__rrrrrr"rrrr.s/3$rrceZdZdZdZdZy) _tqdm_clscNtrtj|i|St|i|Sr) _tqdm_activetqdm_libtqdmrrs r__call__z_tqdm_cls.__call__Gs) ==$1&1 1d-f- -rc\d|_trtjj|i|Syr)rBrrrset_lockrs rrz_tqdm_cls.set_lockMs+ ==))4:6: : rcLtrtjjSyr)rrrget_lockrs rrz_tqdm_cls.get_lockRs ==))+ + rN)rr r!rrrr"rrrrFs. ; ,rrc ttS)zCReturn a boolean indicating whether tqdm progress bars are enabled.)boolrr"rris_progress_bar_enabledrZs  rcday)zEnable tqdm progress bar.TNrr"rrenable_progress_barr`s Lrcday)zDisable tqdm progress bar.FNrr"rrdisable_progress_barrfs Lr)r$Nr)>rrr2rD threadingrrrrrrr r rr rdistributed_utilsr LockrBrHandler__annotations__r4r8rr+FilterrLoggerr/intr:strr>r@rLrOdictrQrTrWrZr\r^r`rcrergrkrnrqrsrxrzrrrrrrrr"rrrs    "6  +/'//D(/]] LL ]]   __ LgnnL ,W^^,, C  "3"2'..26(  T#s(^ S4Z 7>>  :s:(3S3T3& "   ?<33T36GOO6610 ( #"!/0,,"{  r