1;jiY v d Z ddlmZ ddlmZmZ ddlmZmZ ddl Z
ddl m Z m
Z
defd Z G d
d e ZdS ) z
Generic utilities
) OrderedDict)fields is_dataclass)AnyTupleN )is_torch_availableis_torch_versionreturnc t rddl}t | |j rdS t | t j S )z;
Tests if `x` is a `torch.Tensor` or `np.ndarray`.
r NT)r torch
isinstanceTensornpndarray)xr
s Q/root/voice-cloning/.venv/lib/python3.11/site-packages/diffusers/utils/outputs.py is_tensorr sF a & & 4 a $ $$ c e Zd ZdZddZddZd Zd Zd Zd Z d
e
de
fd Z d e
d
e
ddf fd Z fdZ
fdZdee
df fdZ xZS )
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 & t rddlt dd r6j j j j j fd dS j j j j j fd 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 Nz.BaseOutput.__init_subclass__..C . CC,g,g%+2E2U2UV\^e2f2f,g,g r c P di j j | | S r r r! s r r% z.BaseOutput.__init_subclass__..I r& r )r torch.utils._pytreer
r r _register_pytree_node
_dict_flattenregister_pytree_node)r$ r
s `@r __init_subclass__z BaseOutput.__init_subclass__6 s & & & & U++
#99K'5ggggg #88K'5ggggg r c t }t | s t j j d t |d j }t fd|dd D }|r6t |t r!|
D ]
\ }}| |< d S |D ]#}t |j }|
| |j < $d S )Nz has no fields.r c 3 F K | ]}t |j d u V d S Ngetattrname.0fieldselfs r z+BaseOutput.__post_init__..T s4 #d#d%GD%*$=$=$E#d#d#d#d#d#dr r ) r len
ValueError __class____name__r1 r2 allr dictitems)r6 class_fields first_fieldother_fields_are_nonekeyvaluer5 vs ` r
__post_init__zBaseOutput.__post_init__L s d|| < J 7 H H HII I dLO$899 ##d#d#d#dS_`a`b`bSc#d#d#d d d )Z T%B%B ) ) / / 1 1
"
"
U !S
"
" &
)
)D%*--='(D$
)
)r c < t d| j j d )Nz$You cannot use ``__delitem__`` on a
instance. Exceptionr: r; r6 argskwargss r __delitem__zBaseOutput.__delitem___ s! bt~?Vbbbcccr c < t d| j j d )Nz#You cannot use ``setdefault`` on a rG rH rJ s r
setdefaultzBaseOutput.setdefaultb s! adn>Uaaabbbr c < t d| j j d )Nz You cannot use ``pop`` on a rG rH rJ s r popzBaseOutput.pope s! Zt~7NZZZ[[[r c < t d| j j d )Nz You cannot use ``update`` on a rG rH rJ s r updatezBaseOutput.updateh s! ]$.:Q]]]^^^r kc t |t r)t | }|| S | | S r/ )r strr= r> to_tuple)r6 rT
inner_dicts r __getitem__zBaseOutput.__getitem__k sF a & djjll++J a= ==??1% %r r2 rC c || v r$|"t || t || d S r/ )keyssuper __setitem__ __setattr__)r6 r2 rC r: s r r^ zBaseOutput.__setattr__r sU 499;; 5#4 GG e , , ,
D%(((((r c t || t || d S r/ )r\ r] r^ )r6 rB rC r: s r r] zBaseOutput.__setitem__x s=
C'''
C'''''r c t s t S t ^}}}t fdt D }||g|R S )Nc 3 B K | ]}t |j V d S r/ r0 r3 s r r7 z(BaseOutput.__reduce__.. s/ II5WT5:..IIIIIIr )r r\
__reduce__tupler )r6 callable_args remainingrK r: s ` r rb zBaseOutput.__reduce__~ s D!! (77%%'' '&+gg&8&8&:&:#%)IIIIF4LLIIIII) )))r .c ^ t fd D S )za
Convert self to a tuple containing all the attributes/keys that are not `None`.
c 3 ( K | ] }| V
d S r/ r )r4 rT r6 s r r7 z&BaseOutput.to_tuple.. s' 22T!W222222r )rc r[ )r6 s `r rW zBaseOutput.to_tuple s0 2222diikk222222r )r N)r;
__module__ __qualname____doc__r, rE rM rO rQ rS r rY r^ r] rb r rW
__classcell__)r: s @r r r ( sC ,) ) ) )&d d dc c c\ \ \_ _ _&S &S & & & &) )C )D ) ) ) ) ) ) ( ( ( ( ( * * * * *3%S/ 3 3 3 3 3 3 3 3r r )rk collectionsr dataclassesr r typingr r numpyr import_utilsr r
boolr r r r r rs s $ # # # # # , , , , , , , , > > > > > > > >
%D
%
%
%
%a3 a3 a3 a3 a3 a3 a3 a3 a3 a3r