
    Z js                         S SK Jr  S SKJr  SSKJrJr  SSKJr  SSK	J
r
  SSKJr  \
" S	S
9\ " S S\\5      5       5       r\
" S	S
9\ " S S\5      5       5       rSS/rg)    )Sequence)strict   )BackboneConfigMixin%consolidate_backbone_kwargs_to_config)PreTrainedConfig)auto_docstring   )
AutoConfigzPaddlePaddle/UVDoc_safetensors)
checkpointc                     ^  \ rS rSr% SrSrSr\\   S-  \	S'   Sr
\\   S-  \	S'   Sr\\\   \\S4   -     \	S	'   S
r\\\\\\\4   \\\-     -        \	S'   Sr\\\\S4   \\   -        \	S'   Sr\\	S'   U 4S jrSrU =r$ )UVDocBackboneConfig    al  
resnet_head (`Sequence[list[int] | tuple[int, ...]]`, *optional*, defaults to `((3, 32), (32, 32))`):
    Configuration for the ResNet head layers in format [in_channels, out_channels].
resnet_configs (`Sequence[Sequence[tuple[int, int, int, bool] | list[int | bool]]]`, *optional*, defaults to `(((32, 32, 1, False),
    (32, 32, 3, False), (32, 32, 3, False)), ((32, 64, 1, True), (64, 64, 3, False), (64, 64, 3, False), (64, 64, 3, False)), ((64, 128, 1, True),
    (128, 128, 3, False), (128, 128, 3, False), (128, 128, 3, False), (128, 128, 3, False), (128, 128, 3, False)))`):
    Configuration for the ResNet stages in format [in_channels, out_channels, dilation_value, downsample].
stage_configs (Sequence[Sequence[tuple[int, ...] | list[int]]], *optional*, defaults to `(((128, 1),), ((128, 2),),
    ((128, 5),), ((128, 8),(128, 3),(128, 2),), ((128, 12), (128, 7), (128, 4),), ((128, 18), (128, 12), (128, 6),),)`):
    Configuration for the bridge module stages in format [in_channels, dilation_value].
    Each inner sequence corresponds to a single bridge block, and the outer sequence groups blocks by bridge stage.
uvdoc_backboneN_out_features_out_indices))r   r   )r   r   .resnet_head)))r   r      Fr   r   r   Fr   ))r   @   r   Tr   r   r   Fr   r   ))r      r   Tr   r   r   Fr   r   r   r   resnet_configs)))r   r   )r   r
   ))r      ))r      )r   r   r   )r      )r      )r      ))r      r   )r      stage_configsr   kernel_sizec                 n  > U R                    Vs/ s H  n[        U5      PM     snU l        S/[        S[        U R                   5      S-   5       Vs/ s H  nSU 3PM
     sn-   U l        U R                  UR                  SS 5      UR                  SS 5      S9  [        TU ]   " S0 UD6  g s  snf s  snf )Nstemr   stageout_indicesout_features)r)   r*    )	r$   lendepthsrangestage_names"set_output_features_output_indicespopsuper__post_init__)selfkwargsstagesidx	__class__s       ~/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/uvdoc/configuration_uvdoc.pyr3   !UVDocBackboneConfig.__post_init__g   s    151C1CD1Cvs6{1CD"8aTM_M_I`cdId@e&f@ese}@e&ff//

=$7fjjQ_aeFf 	0 	
 	'' E&fs   B-B2)r-   r/   )__name__
__module____qualname____firstlineno____doc__
model_typer   liststr__annotations__r   intr   r   tupler   boolr$   r%   r3   __static_attributes____classcell__r8   s   @r9   r   r       s     "J&*M49t#*%)L$s)d"):K$s)eCHo56 
YNHXeCc4,?&@4d
CS&STU ,FM8HU38_tCy%@AB * K( (    r   c                      ^  \ rS rSr% SrSrS\0rSr\	\
-  S-  \S'   Sr\\S'   Sr\\S	'   S
r\\S'   Sr\\   \\S4   -  \S'   Sr\\\   \\S4   -     \S'   U 4S jrSrU =r$ )UVDocConfigp   a  
padding_mode (`str`, *optional*, defaults to `"reflect"`):
    Padding mode for convolutional layers. Supported modes are `"reflect"`, `"constant"`, and `"replicate"`.
kernel_size (`int`, *optional*, defaults to 5):
    Kernel size for convolutional layers in the backbone network.
bridge_connector (`list[int] | tuple[int, ...]`, *optional*, defaults to `(128, 128)`):
    Configuration for the bridge connector in format [in_channels, out_channels].
out_point_positions2D (`Sequence[list[int] | tuple[int, ...]]`, *optional*, defaults to `((128, 32), (32, 2))`):
    Configuration for the output point positions 2D layer in format [in_channels, out_channels].
uvdocbackbone_configNprelu
hidden_actreflectpadding_moder   r%   )r   r   .bridge_connector))r   r   )r   r
   out_point_positions2Dc                 f   > [        SU R                  SS.UD6u  U l        n[        TU ]  " S0 UD6  g )Nr   )rO   default_config_typer+   )r   rO   r2   r3   )r4   r5   r8   s     r9   r3   UVDocConfig.__post_init__   sB    'L (
 00 0(
 (
$f
 	''rJ   )rO   )r;   r<   r=   r>   r?   r@   r   sub_configsrO   dictr   rC   rQ   rB   rS   r%   rD   rT   rA   rE   rU   r   r3   rG   rH   rI   s   @r9   rL   rL   p   s    	 J$j1K6:OT,,t3:J!L#!K4>d3i%S/1>CW8DIc3h$?@W( (rJ   rL   N)collections.abcr   huggingface_hub.dataclassesr   backbone_utilsr   r   configuration_utilsr   utilsr	   autor   r   rL   __all__r+   rJ   r9   <module>rb      s   , % . X 3 #  ;<K(-/? K(  =K(\ ;<(" (  =(> !-
0rJ   