declare-lab/tango2
92
1a
2 ��'de$ � @ s U d Z ddlZddlZddlZddlZddlmZmZmZmZm Z m3Z4m Z m Z ddl
mZ ddlmZ e�� Zdaeej ed< ejej ej ejejd�Zej Zdad d5� Zed �d d
�Z ej d �dd�Z dd �dd�Z dd �dd�Z dd� Z!dCee ej d�dd�Z"e#d �dd�Z$e#dd�d d �Z%d d � Z&d d!� Z'd"d#� Z(d$d%� Z)dd �d&d'�Z*dd �d(d)�Z+ejdd*�d+d,�Z,ejdd*�d-d.�Z-dd �d/d0�Z.dd �d1d2�Z/dd �d3d4�Z0dd �d5d6�Z1d7d8� Z2e2ej _2G d9d:� d:�Z3G d;d<� d<�Z4e4� Ze5d �d=d>�Z6d?d@� Z7dAdB� Z8dS )Dz Logging utilities.� N)�CRITICAL�DEBUG�ERROR�FATAL�INFO�NOTSET�WARN�WARNING)�Optional)�auto�_default_handler)�debug�info�warning�error�criticalTc C sH t �dd�} | rD| tv r t| S t�� �d| � dd�t�� �� �� tS )z�6 If DIFFUSERS_VERBOSITY env var is set to one of the valid choices return that as the new default level. If it is7 not - fall back to `_default_log_level`8 ZDIFFUSERS_VERBOSITYNz#Unknown option DIFFUSERS_VERBOSITY=z, has to be one of: z, ) �os�getenv�9log_levels�logging� getLoggerr �join�keys�_default_log_level)�
env_level_str� r �L/home/deep/Projects/audio_diffusion/diffusers/src/diffusers/utils/logging.py�_get_default_logging_level4 s ��r )�returnc C s t �d�d S )N�.r )�__name__�splitr r r r �_get_library_nameE s r" c C s t �t� �S �N)r r r" r r r r �_get_library_root_loggerI s r$ c C sp t �X trW d � d S t�� atjjt_t� } | �t� | � t10� � d| _ W d � n1 sb0 Y d S )NF) �_lockr r �
StreamHandler�sys�stderr�flushr$ �11addHandler�setLevelr � propagate��library_root_loggerr r r � _configure_library_root_loggerM s 1213 r/ c C s\ t �D tsW d � d S t� } | �t� | �tj� d aW d � n1 sN0 Y d S r# )r% r r$ �
removeHandlerr+ r r r- r r r �_reset_library_root_logger^ s 14 r1 c C s t S r# )r r r r r �get_log_levels_dictk s r2 )�namer c C s | du rt � } t� t�| �S )z�15 Return a logger with the specified name.16 17 This function is not supposed to be directly accessed unless you are writing a custom diffusers module.18 N)r" r/ r r )r3 r r r �19get_loggero s r4 c C s t � t� �� S )u� 20 Return the current level for the 🤗 Diffusers' root logger as an int.21 22 Returns:23 `int`: The logging level.24 25 <Tip>26 27 🤗 Diffusers has following logging levels:28 29 - 50: `diffusers.logging.CRITICAL` or `diffusers.logging.FATAL`30 - 40: `diffusers.logging.ERROR`31 - 30: `diffusers.logging.WARNING` or `diffusers.logging.WARN`32 - 20: `diffusers.logging.INFO`33 - 10: `diffusers.logging.DEBUG`34 35 </Tip>)r/ r$ �getEffectiveLevelr r r r �
get_verbosity} s r6 )� verbosityr c C s t � t� �| � dS )u� 36 Set the verbosity level for the 🤗 Diffusers' root logger.37 38 Args:39 verbosity (`int`):40 Logging level, e.g., one of:41 42 - `diffusers.logging.CRITICAL` or `diffusers.logging.FATAL`43 - `diffusers.logging.ERROR`44 - `diffusers.logging.WARNING` or `diffusers.logging.WARN`45 - `diffusers.logging.INFO`46 - `diffusers.logging.DEBUG`47 N)r/ r$ r+ )r7 r r r �
set_verbosity� s r8 c C s t t�S )z&Set the verbosity to the `INFO` level.)r8 r r r r r �set_verbosity_info� s r9 c C s t t�S )z)Set the verbosity to the `WARNING` level.)r8 r r r r r �set_verbosity_warning� s r: c C s t t�S )z'Set the verbosity to the `DEBUG` level.)r8 r r r r r �set_verbosity_debug� s r; c C s t t�S )z'Set the verbosity to the `ERROR` level.)r8 r r r r r �set_verbosity_error� s r< c C s" t � tdusJ �t� �t� dS )zFDisable the default handler of the HuggingFace Diffusers' root logger.N)r/ r r$ r0 r r r r �disable_default_handler� s r= c C s" t � tdusJ �t� �t� dS )zEEnable the default handler of the HuggingFace Diffusers' root logger.N)r/ r r$ r* r r r r �enable_default_handler� s r> )�handlerr c C s" t � | dusJ �t� �| � dS )z9adds a handler to the HuggingFace Diffusers' root logger.N)r/ r$ r* �r? r r r � add_handler� s rA c C s. t � | dur| t� jvs J �t� �| � dS )zBremoves given handler from the HuggingFace Diffusers' root logger.N)r/ r$ �handlersr0 r@ r r r �remove_handler� s rC c C s t � dt� _dS )zk48 Disable propagation of the library log outputs. Note that log propagation is disabled by default.49 FN�r/ r$ r, r r r r �disable_propagation� s rE c C s t � dt� _dS )z�50 Enable propagation of the library log outputs. Please disable the HuggingFace Diffusers' default handler to prevent51 double logging if the root logger has been configured.52 TNrD r r r r �enable_propagation� s rF c C s* t � j} | D ]}t�d�}|�|� q dS )a
53 Enable explicit formatting for every HuggingFace Diffusers' logger. The explicit formatter is as follows:54 ```55 [LEVELNAME|FILENAME|LINE NUMBER] TIME >> MESSAGE56 ```57 All handlers currently bound to the root logger are affected by this method.58 zB[%(levelname)s|%(filename)s:%(lineno)s] %(asctime)s >> %(message)sN)r$ rB r � Formatter� setFormatter)rB r? � formatterr r r �enable_explicit_format� s 59rJ c C s t � j} | D ]}|�d� q dS )z�60 Resets the formatting for HuggingFace Diffusers' loggers.61 62 All handlers currently bound to the root logger are affected by this method.63 N)r$ rB rH )rB r? r r r � reset_format s rK c O s( t �dd�}|rdS | j|i |�� dS )z�64 This method is identical to `logger.warning()`, but if env var DIFFUSERS_NO_ADVISORY_WARNINGS=1 is set, this65 warning will not be printed66 Z DIFFUSERS_NO_ADVISORY_WARNINGSFN)r r r )�self�args�kwargsZno_advisory_warningsr r r �warning_advice
s rO c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd67d � Zd S )
� EmptyTqdmz%Dummy tqdm which doesn't do anything.c O s |r |d nd | _ d S )Nr )� _iterator�rL rM rN r r r �__init__ s zEmptyTqdm.__init__c C s68 t | j�S r# )�iterrQ �rL r r r �__iter__! s zEmptyTqdm.__iter__c C s dd� }|S )zReturn empty function.c _ s d S r# r )rM rN r r r �empty_fn' s z'EmptyTqdm.__getattr__.<locals>.empty_fnr )rL �_rW r r r � __getattr__$ s zEmptyTqdm.__getattr__c C s | S r# r rU r r r � __enter__, s zEmptyTqdm.__enter__c C s d S r# r )rL �type_�value� tracebackr r r �__exit__/ s zEmptyTqdm.__exit__N) r �69__module__� __qualname__�__doc__rS rV rY rZ r^ r r r r rP s rP c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _tqdm_clsc O s& t rtj|i |��S t|i |��S d S r# )� _tqdm_active�tqdm_lib�tqdmrP rR r r r �__call__4 s z_tqdm_cls.__call__c O s d | _ tr tjj|i |��S d S r# )r% rc rd re �set_lockrR r r r rg : s z_tqdm_cls.set_lockc C s t rtj�� S d S r# )rc rd re �get_lockrU r r r rh ? s z_tqdm_cls.get_lockN)r r_ r` rf rg rh r r r r rb 3 s rb c C s t t�S )zCReturn a boolean indicating whether tqdm progress bars are enabled.)�boolrc r r r r �is_progress_bar_enabledG s rj c C s da dS )zEnable tqdm progress bar.TN�rc r r r r �enable_progress_barM s rl c C s da dS )zDisable tqdm progress bar.FNrk r r r r �disable_progress_barS s rm )N)9ra r r r' � threadingr r r r r r r r �typingr70 re r rd �Lockr% r �Handler�__annotations__r r rc r �strr" �Loggerr$ r/ r1 r2 r4 �intr6 r8 r9 r: r; r< r= r> rA rC rE rF rJ rK rO rP rb ri rj rl rm r r r r �<module> s\ (71 �
72