+
    Lj_	                     L    ^ RI t  ! R R] P                  P                  4      tR# )    Nc                   >   a a ] tR t^t oV3R lV 3R lltRtVtV ;t# )AttnProcsLayersc                J   < V ^8  d   QhRS[ S[S[P                  3,          /# )   
state_dict)dictstrtorchTensor)format__classdict__s   "C/app/.local/lib/python3.14/site-packages/diffusers/loaders/utils.py__annotate__AttnProcsLayers.__annotate__   s%     &L &L4U\\(9#: &L    c                  <a a \         SS `  4        \        P                  P	                  VP                  4       4      S n        \        \        VP                  4       4      4      S n
        \        VP                  4       4       UUu/ uF  w  r#W2bK	  	  uppS n        R R.S n        R pV 3R loV3R lpS P                  V4       S P                  VRR7       R# u uppi )z
.processorz
.self_attnc                     / pVP                  4        FP  w  rV\        VP                  R 4      ^,          4      pVP                  RV 2V P                  V,          4      pWdV&   KR  	  V# ).layers.)itemsintsplitreplacemapping)	moduler   argskwargsnew_state_dictkeyvaluenumnew_keys	   &&*,     r   map_to(AttnProcsLayers.__init__.<locals>.map_to   sb    N(..0
#))C.+,++uov~~c7JK*/w' 1
 "!r   c           	         < SP                    F*  pW 9   g   K  V P                  V4      ^ ,          V,           u # 	  \        R\        VP	                  4       4       RV  RSP                    R24      h)r   z1There seems to be a problem with the state_dict: z. z has to have one of r   )
split_keysr   
ValueErrorsetkeys)r   r   kselfs   && r   	remap_key+AttnProcsLayers.__init__.<locals>.remap_key(   s    __899Q<?Q.. % CC
HYDZC[[]^a]bbvw{  xG  xG  wH  HI  J r   c                    < \        VP                  4       4      pV F=  pS! WQ4      pVP                  VR V P                  V,           24      pW,          W&   W K?  	  R# )r   N)listr)   r   rev_mapping)	r   r   r   r   all_keysr   replace_keyr"   r,   s	   &&*,    r   map_from*AttnProcsLayers.__init__.<locals>.map_from1   s[    JOO-.H'8++kWV=O=OP[=\<]3^_&0o
#O	  r   T)with_moduleN)super__init__r
   nn
ModuleListvalueslayersr   	enumerater)   r   r0   r&   _register_state_dict_hook"_register_load_state_dict_pre_hook)r+   r   r*   vr#   r3   r,   	__class__s   f&    @r   r7   AttnProcsLayers.__init__   s    hh))**;*;*=>Ijoo&789-6z7H-IJ-ITQAD-IJ (6	"		$ 	&&v.//d/KE Ks   C)r;   r   r0   r&   )__name__
__module____qualname____firstlineno__r7   __static_attributes____classdictcell____classcell__)r@   r   s   @@r   r   r      s     &L &L &Lr   r   )r
   r8   Moduler    r   r   <module>rK      s      'Lehhoo 'Lr   