o
piY @ sj d Z ddlmZ ddlmZmZ ddlmZmZ ddl Z
ddl m Z m
Z
defd d
ZG d d d eZdS )
z
Generic utilities
) OrderedDict)fields is_dataclass)AnyTupleN )is_torch_availableis_torch_versionreturnc C s* t rddl}t| |jrdS t| tjS )z;
Tests if `x` is a `torch.Tensor` or `np.ndarray`.
r NT)r torch
isinstanceTensornpndarray)xr r U/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/utils/outputs.py is_tensor s
r c s e Zd ZdZd ddZd ddZdd Zd
d Zd d
Zdd Z de
de
fddZ de
de
ddf fdd Z fddZ
fddZdee
df fd d Z ZS )
BaseOutputa
Base class for all model outputs as dataclass. Has a `__getitem__` that allows indexing by integer or slice (like a
tuple) or strings (like a dictionary) that will ignore the `None` attributes. Otherwise behaves like a regular
Python dictionary.
You can't unpack a [`BaseOutput`] directly. Use the [`~utils.BaseOutput.to_tuple`] method to convert it to a tuple
first.
r
Nc sh t r2ddltddr jj jjj fdd dS jj jjj fdd dS dS )zRegister subclasses as pytree nodes.
This is necessary to synchronize gradients when using `torch.nn.parallel.DistributedDataParallel` with
`static_graph=True` with modules that output `ModelOutput` subclasses.
r NC z.BaseOutput.__init_subclass__..c r r r r r r r r! I r" )r torch.utils._pytreer r r _register_pytree_node
_dict_flattenregister_pytree_node)r r r r __init_subclass__6 s
z BaseOutput.__init_subclass__c s t }t|st jj dt |d j}t fdd|dd D }|r>t|t r>|
D ]\}}| |< q3d S |D ]}t |j}|d urQ| |j< q@d S )Nz has no fields.r c 3 s | ] }t |jd u V qd S Ngetattrname.0fieldselfr r T s z+BaseOutput.__post_init__..r ) r len
ValueError __class____name__r* r+ allr dictitems)r0 class_fields first_fieldother_fields_are_nonekeyvaluer. vr r/ r
__post_init__L s
zBaseOutput.__post_init__c O t d| jj d)Nz$You cannot use ``__delitem__`` on a
instance. Exceptionr4 r5 r0 argskwargsr r r __delitem___ zBaseOutput.__delitem__c O r@ )Nz#You cannot use ``setdefault`` on a rA rB rD r r r
setdefaultb rH zBaseOutput.setdefaultc O r@ )Nz You cannot use ``pop`` on a rA rB rD r r r pope rH zBaseOutput.popc O r@ )Nz You cannot use ``update`` on a rA rB rD r r r updateh rH zBaseOutput.updatekc C s* t |trt| }|| S | | S r( )r strr7 r8 to_tuple)r0 rL
inner_dictr r r __getitem__k s
zBaseOutput.__getitem__r+ r= c s4 || v r|d urt || t || d S r( )keyssuper __setitem__ __setattr__)r0 r+ r= r4 r r rT r s zBaseOutput.__setattr__c s t || t || d S r( )rR rS rT )r0 r< r= rU r r rS x s zBaseOutput.__setitem__c sJ t s t S t ^}}}t fddt D }||g|R S )Nc 3 s | ] }t |jV qd S r( r) r, r/ r r r1 s z(BaseOutput.__reduce__..)r rR
__reduce__tupler )r0 callable_args remainingrE rU r/ r rV ~ s
zBaseOutput.__reduce__.c s t fdd D S )za
Convert self to a tuple containing all the attributes/keys that are not `None`.
c 3 s | ]} | V qd S r( r )r- rL r/ r r r1 s z&BaseOutput.to_tuple..)rW rQ r/ r r/ r rN s zBaseOutput.to_tuple)r
N)r5
__module__ __qualname____doc__r' r? rG rI rJ rK r rP rT rS rV r rN
__classcell__r r rU r r ( s
r )r] collectionsr dataclassesr r typingr r numpyr import_utilsr r boolr r r r r r s