
    3jX                     h    S SK Jr  S SKJr  SSKJr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ModelTilingSizeRequirements	StateDict   )CIDNetc                   N   ^  \ rS rSrSU 4S jjr\S\S\\   4S j5       r	Sr
U =r$ )HVICIDNetArch   returnc           #         > [         TU ]  SS[        R                  " SSSSSSS	S
SSSSSSSSSSSSSSSSSSSSS5      S 9  g )!N	HVICIDNetz
HVI-CIDNetzHVE_block0.1.weightzHVE_block1.prelu.weightHVE_block1.down.0.weightHVE_block3.down.0.weightzHVD_block3.prelu.weightzHVD_block3.up_scale.0.weightzHVD_block3.up.weightzHVD_block1.up.weightzHVD_block0.1.weightzIE_block0.1.weightzIE_block1.prelu.weightzIE_block1.down.0.weightzID_block1.up.weightzID_block0.1.weightzHV_LCA1.gdfn.project_in.weightzHV_LCA1.gdfn.dwconv.weightzHV_LCA1.gdfn.dwconv1.weightzHV_LCA1.gdfn.dwconv2.weightzHV_LCA1.gdfn.project_out.weightzHV_LCA1.norm.weightHV_LCA1.ffn.temperaturezHV_LCA1.ffn.q.weightzHV_LCA1.ffn.q_dwconv.weightzHV_LCA1.ffn.project_out.weightzHV_LCA2.gdfn.project_in.weightzHV_LCA6.gdfn.project_in.weightzI_LCA1.gdfn.project_in.weightzI_LCA6.ffn.project_out.weightztrans.density_k)idnamedetect)super__init__r   has_all)self	__class__s    c/home/wildlama/miniconda3/lib/python3.13/site-packages/spandrel/architectures/HVICIDNet/__init__.pyr   HVICIDNetArch.__init__   s    ''%)**).&&%$()%$0,--1%)&-000//!; 	 "	
    
state_dictc                    / SQn/ SQnSnUS   R                   S   US   R                   S   US   R                   S   US   R                   S   /nSUS	   R                   S   US
   R                   S   US   R                   S   /nSU;   n[        UUUS9n[        UUU S/ SSSSS[        SS9[        R
                  S9$ )N)$   r#   H      )r            Fr   r   r   zHVE_block2.down.0.weightr   r   zHV_LCA2.ffn.temperaturezHV_LCA3.ffn.temperaturezHVE_block1.norm.weight)channelsheadsnormRestorationTr   r(   )multiple_of)
architecturepurposetagssupports_halfsupports_bfloat16scaleinput_channelsoutput_channelssize_requirementstiling)shaper   r   r	   r   DISCOURAGED)r   r!   r)   r*   r+   models         r   loadHVICIDNetArch.load5   s   $ 1288;1288;1288;1288;	
 0177:0177:0177:	
 (:5
 $!".1=**
 	
r     )r   N)__name__
__module____qualname____firstlineno__r   r   r
   r   r   r;   __static_attributes____classcell__)r   s   @r   r   r      s1    #
J (
y (
-A)-L (
 (
r    r   r   N)typing_extensionsr   spandrel.utilr   __helpers.model_descriptorr   r   r   r	   r
   __arch.cidnetr   r   r   __all__r=   r    r   <module>rI      s:    & &  /O
L+ O
d K
(r    