bi n d Z ddlmZ ddlmZmZ ddlmZmZ ddl Z
ddl m Z m
Z
defd Z G d
d e Zy) z
Generic utilities
) OrderedDict)fields is_dataclass)AnyTupleN )is_torch_availableis_torch_versionreturnc t rddl}t | |j ryt | t j
S )z;
Tests if `x` is a `torch.Tensor` or `np.ndarray`.
r NT)r torch
isinstanceTensornpndarray)xr
s R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/utils/outputs.py is_tensorr s/ a & 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 rJj j j j j j fd yj j j j j j fd j d j yy) 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 ( C,g%++2E2E2U2UV\^e2f,g r c \ di j j j | | S r r r! s r r% z.BaseOutput.__init_subclass__..I r& r .)serialized_type_name)
r torch.utils._pytreer
r r _register_pytree_node
_dict_flattenregister_pytree_node
__module____name__)r$ r
s `@r __init_subclass__z BaseOutput.__init_subclass__6 s & U+ ##99KK''55g ##88KK''55g,/NN+;1S\\N)K 9 r c t }t | s"t j j d t |d j }t fd|dd D }|r.t |t r |j D ]
\ }}| |< y |D ]* }t |j }| | |j < , y )Nz has no fields.r c 3 N K | ] }t |j d u y wNgetattrname.0fieldselfs r z+BaseOutput.__post_init__..U s! #d%GD%**$=$E#ds "%r ) r len
ValueError __class__r/ r5 r6 allr dictitems)r: class_fields first_fieldother_fields_are_nonekeyvaluer9 vs ` r
__post_init__zBaseOutput.__post_init__M s d| < 7 7 8 HI I dLO$8$89 ##dS_`a`bSc#d d Z T%B ) / / 1
"
U !S
" &
)D%**-='(D$
)r c H t d| j j d )Nz$You cannot use ``__delitem__`` on a
instance. Exceptionr> r/ r: argskwargss r __delitem__zBaseOutput.__delitem__` s# >t~~?V?V>WWabccr c H t d| j j d )Nz#You cannot use ``setdefault`` on a rJ rK rM s r
setdefaultzBaseOutput.setdefaultc s# =dnn>U>U=VV`abbr c H t d| j j d )Nz You cannot use ``pop`` on a rJ rK rM s r popzBaseOutput.popf s" 6t~~7N7N6OzZ[[r c H t d| j j d )Nz You cannot use ``update`` on a rJ rK rM s r updatezBaseOutput.updatei s# 9$..:Q:Q9RR\]^^r kc t |t r t | j }|| S | j | S r3 )r strr@ rA to_tuple)r: rW
inner_dicts r __getitem__zBaseOutput.__getitem__l s7 a djjl+J a= ==?1% %r r6 rF c n || j v r|t | || t |
|| y r3 )keyssuper __setitem__ __setattr__)r: r6 rF r> s r ra zBaseOutput.__setattr__s s4 499; 5#4 G e ,
D%(r c F t | || t | || y r3 )r_ r` ra )r: rE rF r> s r r` zBaseOutput.__setitem__y s!
C'
C'r c t st S t ^}}}t fdt D }||g|S )Nc 3 J K | ] }t |j y wr3 r4 r7 s r r; z(BaseOutput.__reduce__.. s I5WT5::.Is #)r r_
__reduce__tupler )r: callable_args remainingrN r> s ` r re zBaseOutput.__reduce__ sQ D!7%' '&+g&8&:#%)IF4LII) ))r .c H t fd j D S )za
Convert self to a tuple containing all the attributes/keys that are not `None`.
c 3 ( K | ] }| y wr3 r )r8 rW r: s r r; z&BaseOutput.to_tuple.. s 2T!W2s )rf r^ )r: s `r rZ zBaseOutput.to_tuple s 2diik222r )r N)r/ r. __qualname____doc__r0 rH rP rR rT rV r r\ ra r` re r rZ
__classcell__)r> s @r r r ( su .)&dc\_&S &S &) )C )D ) ( *3%S/ 3r r )rm collectionsr dataclassesr r typingr r numpyr import_utilsr r
boolr r r r r ru s5 $ , >
%D
%b3 b3r