
    3j	                     d    S SK Jr  S SK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   )ArchitectureImageModelDescriptorSizeRequirements	StateDict   )GFPGANv1Cleanc                   N   ^  \ rS rSrSU 4S jjr\S\S\\   4S j5       r	Sr
U =r$ )
GFPGANArch   returnc                 L   > [         TU ]  S[        R                  " SS5      S9  g )NGFPGANztoRGB.0.weightz#stylegan_decoder.style_mlp.1.weight)iddetect)super__init__r   has_all)self	__class__s    `/home/wildlama/miniconda3/lib/python3.13/site-packages/spandrel/architectures/GFPGAN/__init__.pyr   GFPGANArch.__init__   s,    '' 5 	 	
    
state_dictc                     SnSnSnS nSnSnSnSn	Sn
Sn[        UUUUUUUU	U
US9
n[        UUU S/ SSSS	S	[        SS
9S S9$ )Ni      F   Tr
   )
out_sizenum_style_featchannel_multiplierdecoder_load_pathfix_decodernum_mlpinput_is_latentdifferent_wnarrowsft_halfFaceSRr   )minimumc                     U " U5      S   $ )Nr    )modelimages     r   <lambda>!GFPGANArch.load.<locals>.<lambda>>   s    uar   )
architecturepurposetagssupports_halfsupports_bfloat16scaleinput_channelsoutput_channelssize_requirementscall_fn)r   r   r   )r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r.   s                r   loadGFPGANArch.load   s     )1/#+#
 $".s;8
 	
r   r-   )r   N)__name__
__module____qualname____firstlineno__r   r   r	   r   r   r<   __static_attributes____classcell__)r   s   @r   r   r      s0    
 &
y &
-A&-I &
 &
r   r   r   N)typing_extensionsr   spandrel.utilr   __helpers.model_descriptorr   r   r   r	   __arch.gfpganv1_clean_archr   r   r   __all__r-   r   r   <module>rI      s9    & &  @1
f% 1
h 
"r   