
    3j                     h    S SK Jr  S SKJrJr  SSKJrJrJrJ	r	  SSK
Jr   " S S\\   5      rSS	/rg
)    )override)KeyConditionget_seq_len   )ArchitectureMaskedImageModelDescriptorSizeRequirements	StateDict   )LaMac                   N   ^  \ rS rSrSU 4S jjr\S\S\\   4S j5       r	Sr
U =r$ )LaMaArch   returnc                 L   > [         TU ]  S[        R                  " SS5      S9  g )Nr   zmodel.model.1.bn_l.running_meanz#generator.model.1.bn_l.running_mean)iddetect)super__init__r   has_any)self	__class__s    ^/home/wildlama/miniconda3/lib/python3.13/site-packages/spandrel/architectures/LaMa/__init__.pyr   LaMaArch.__init__   s,    ''15 	 	
    
state_dictc                 <   UR                  5        VVs0 s H  u  p#UR                  SS5      U_M     nnnSnSnUS   R                  S   n[        US5      nUSUS-
   S3   R                  S	   n[	        UUS
9n[        UUU S/ SSUS-
  U[        SSS9S9
$ s  snnf )Nzgenerator.modelzmodel.model   r   z model.model.1.ffc.convl2l.weightr   zmodel.model.z.weightr   )in_ncout_nc
InpaintingFT      )minimummultiple_of)architecturepurposetagssupports_halfsupports_bfloat16input_channelsoutput_channelssize_requirements)itemsreplaceshaper   r   r   r	   )r   r   kvr   r    seq_lenmodels           r   loadLaMaArch.load   s     #((*
* II'7:* 	 

 =>DDQGj-8l7Q;-w?@FFqI

 * " 19".rqI
 	
%
s   B )r   N)__name__
__module____qualname____firstlineno__r   r   r
   r   r   r5   __static_attributes____classcell__)r   s   @r   r   r      s0    
 
y 
-G-M 
 
r   r   r   N)typing_extensionsr   spandrel.utilr   r   __helpers.model_descriptorr   r   r	   r
   __arch.LaMar   r   __all__r7   r   r   <module>rC      s8    & 3  )
|D! )
X v
r   