
    IЦi              	          % S SK Jr  S SKJrJrJrJrJrJrJ	r	J
r
  S SKrS SKJrJr  \\\/\4   r\\\/\4   r0 r\\
\   \4   \S'   0 r\\
\   \\   4   \S'    SS\
\   S\S	\\   S
S4S jjr SS\S\S
\\   4S jjrS\\\4   S\S
\\   4S jrS\\   S\S
\\   4S jrS\	\   S\S
\\   4S jrS\S\S
\\   4S jrS\\\4   S\S
\4S jrS\\   S\S
\4S jrS\	\   S\S
\4S jrS\S\S
\4S jr\" \ \\5        \" \!\\5        \" \"\\5        \RF                  RH                   H  r%\" \%\\5        M     \" \\\5        g)    )
namedtuple)AnyCallableDictList
NamedTupleOptionalTupleTypeN)PyTreeTreeSpecSUPPORTED_NODESSUPPORTED_NODES_EXACT_MATCHclsflatten_fn_specflatten_fn_exact_match_specreturnc                 (    U[         U '   U[        U '   g N)r   r   )r   r   r   s      O/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/fx/_pytree.pyregister_pytree_flatten_specr      s    
 +OC'B$    pytreespecc                    UR                  5       (       a  U /$ UR                  [        ;  a  [        [        U 5       S35      e[        UR                     nU" X5      nU(       a9  [        UR                     nU(       a  U" U U5      (       d  [        SU  SU 35      e/ n[        XAR                  5       H  u  px[        XxU5      n	Xi-  nM     U$ )Nz does not have a flatten_fn_spec associated with it. Please register one with torch.fx._pytree.register_pytree_flatten_spec.  If you have serialized your model, make sure that any custom pytrees have been registered before loading it.zCannot flatten pytree z, given spec: )is_leaftyper   RuntimeErrorr   zipchildren_specstree_flatten_spec)
r   r   exact_structural_matchr   child_pytreesr   resultchild
child_specflats
             r   r!   r!      s    
 ||~~xyy'F|n S S
 	

 &dii0O#F1M&A$))&L#&/J0
 0
 !7x~dVTUUF 0C0CD 4JK E Mr   dc                 J    UR                    Vs/ s H  o U   PM	     sn$ s  snf r   )context)r(   r   ks      r   _dict_flatten_specr,   6   s     ,,',QaD,'''s    c                 \    [        UR                  5       Vs/ s H  o U   PM	     sn$ s  snf r   rangenum_childrenr(   r   is      r   _list_flatten_specr3   :   )     1 1232QaD2333   )c                 \    [        UR                  5       Vs/ s H  o U   PM	     sn$ s  snf r   r.   r1   s      r   _tuple_flatten_specr7   >   r4   r5   c                 \    [        UR                  5       Vs/ s H  o U   PM	     sn$ s  snf r   r.   r1   s      r   _namedtuple_flatten_specr9   B   r4   r5   c                 2    [        U 5      UR                  :H  $ r   lenr0   r(   r   s     r   _dict_flatten_spec_exact_matchr>   F       q6T&&&&r   c                 2    [        U 5      UR                  :H  $ r   r;   r=   s     r   _list_flatten_spec_exact_matchrA   J   r?   r   c                 2    [        U 5      UR                  :H  $ r   r;   r=   s     r   _tuple_flatten_spec_exact_matchrC   N   r?   r   c                 2    [        U 5      UR                  :H  $ r   r;   r=   s     r   $_namedtuple_flatten_spec_exact_matchrE   R   r?   r   r   )F)&collectionsr   typingr   r   r   r   r   r	   r
   r   torch.return_typestorchtorch.utils._pytreer   r   FlattenFuncSpecboolFlattenFuncExactMatchSpecr   __annotations__r   r   r!   r,   r3   r7   r9   r>   rA   rC   rE   dictlisttuplereturn_typesall_return_typesreturn_type r   r   <module>rV      sK   " O O O  0 FH-t34$fh%7%=> 46d3i01 6TV T$s)X6O-P"PQ V HLC	cC$C "**C!DC 
	C !
 
#Y	:($sCx. ( (T#Y (4$s) 48 4S	 445: 4X 4$s) 44
 4( 4tCy 4'd38n 'H ' ''d3i 'x 'D ''uSz ' 'd ''J 'h '4 ' T#57U V T#57U V 	#
 %%66K ' 7 (r   