declare-lab/tango2
92
1a
2 ��'d9 � @ s\ d Z ddlmZ ddlmZ ddlmZmZ ddlZ ddl3m Z dd � Z G d4d � d e�Z
dS ) z5Generic utilities6� )� OrderedDict)�fields)�Any�TupleN� )�is_torch_availablec C s* t � r ddl}t| |j�r dS t| tj�S )z;7 Tests if `x` is a `torch.Tensor` or `np.ndarray`.8 r NT)r �torch�9isinstance�Tensor�np�ndarray)�xr � r �L/home/deep/Projects/audio_diffusion/diffusers/src/diffusers/utils/outputs.py� is_tensor s10 r c sn e Zd ZdZdd� Zdd� Zdd� Zdd � Zd11d � Zd d
� Z � fdd�Z12� fdd�Z e e
d�dd�Z� ZS )�13BaseOutputa� 14 Base class for all model outputs as dataclass. Has a `__getitem__` that allows indexing by integer or slice (like a15 tuple) or strings (like a dictionary) that will ignore the `None` attributes. Otherwise behaves like a regular16 python dictionary.17 18 <Tip warning={true}>19 20 You can't unpack a `BaseOutput` directly. Use the [`~utils.BaseOutput.to_tuple`] method to convert it to a tuple21 before.22 23 </Tip>24 c s� t � �}t|�s"t� jj� d���t� |d j�}t� fdd�|dd � D ��}|rzt|t �rz|�25� D ]\}}|� |< qfn(|D ]"}t� |j�}|d ur~|� |j< q~d S )Nz has no fields.r c 3 s | ]}t � |j�d u V qd S �N)�getattr�name)�.0�field��selfr r � <genexpr>> � z+BaseOutput.__post_init__.<locals>.<genexpr>r ) r �len�26ValueError� __class__�__name__r r �allr �dict�items)r � class_fields� first_field�other_fields_are_none�key�valuer �vr r r �
__post_init__6 s zBaseOutput.__post_init__c O s t d| jj� d���d S )Nz$You cannot use ``__delitem__`` on a �27 instance.�� Exceptionr r �r �args�kwargsr r r � __delitem__I s zBaseOutput.__delitem__c O s t d| jj� d���d S )Nz#You cannot use ``setdefault`` on a r) r* r, r r r �28setdefaultL s zBaseOutput.setdefaultc O s t d| jj� d���d S )Nz You cannot use ``pop`` on a r) r* r, r r r �popO s zBaseOutput.popc O s t d| jj� d���d S )Nz You cannot use ``update`` on a r) r* r, r r r �updateR s zBaseOutput.updatec C s. t |t�r t| �� �}|| S | �� | S d S r )r �strr r! �to_tuple)r �k�29inner_dictr r r � __getitem__U s 30 zBaseOutput.__getitem__c s4 || � � v r"|d ur"t� �||� t� �||� d S r )�keys�super� __setitem__� __setattr__)r r r& �r r r r; \ s zBaseOutput.__setattr__c s t � �||� t � �||� d S r )r9 r: r; )r r% r& r< r r r: b s zBaseOutput.__setitem__)�returnc s t � fdd�� �� D ��S )za31 Convert self to a tuple containing all the attributes/keys that are not `None`.32 c 3 s | ]}� | V qd S r r )r r5 r r r r l r z&BaseOutput.to_tuple.<locals>.<genexpr>)�tupler8 r r r r r4 h s zBaseOutput.to_tuple)r �33__module__� __qualname__�__doc__r( r/ r0 r1 r2 r7 r; r: r r r4 �
__classcell__r r r< r r ( s
r )rA � collectionsr � dataclassesr �typingr r �numpyr � import_utilsr r r r r r r �<module> s