iy*dZddlmZddlmZddlmZddlmZddl m Z m Z ed eGd d eZ ed eGd deZ ed eGddeZed eGddeZed eGddeZed eGddeZed eGddeZgdZy)zSAM3 model configuration)strict)CLIPTextConfig)PreTrainedConfig)auto_docstring)CONFIG_MAPPING AutoConfigz facebook/sam3) checkpointceZdZUdZdZdZdZeed<dZ eed<dZ eed <d Z eed <d Z eed <dZ eeezeeefzed<dZeeezeeefzed<dZeed<dZeed<dZeezed<dZeed<dZeed<dZeedzed<dZedzed<dZeeezeeefzed <dZeezed!<d"Zeed#<fd$ZxZS)% Sam3ViTConfiga8 rope_theta (`float`, *optional*, defaults to 10000.0): Base frequency for RoPE. window_size (`int`, *optional*, defaults to 24): Window size for windowed attention. global_attn_indexes (`list[int]`, *optional*, defaults to `[7, 15, 23, 31]`): Indexes of layers with global attention. pretrain_image_size (`int`, *optional*, defaults to 336): Pretrained model image size for position embedding initialization. hidden_dropout (`float`, *optional*, defaults to 0.0): Dropout probability for hidden states. backbone_configsam3_vit_model hidden_sizeiintermediate_size num_hidden_layersnum_attention_headsr num_channelsi image_size patch_sizegelu hidden_actư>layer_norm_epsattention_dropoutg@ rope_theta window_sizeNglobal_attn_indexeslayer_scale_init_valueiPpretrain_image_sizehidden_dropout{Gz?initializer_rangec Pt|di||j gd|_yy)N))super __post_init__r$selfkwargs __class__s |/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/sam3/configuration_sam3.pyr1zSam3ViTConfig.__post_init__>s, ''  # # +'6D $ ,)__name__ __module__ __qualname____doc__base_config_key model_typerint__annotations__rrrrrlisttuplerrstrrfloatr r!r#r$r%r&r'r)r1 __classcell__r5s@r6r r s$ (O!JK!s!s!!L#48Jd3i%S/1846Jd3i%S/16J NE %(us{(JK,0cT)0+/EDL/=@tCy5c?:@"%NECK%#u#77r7r ceZdZUdZdZdZdeiZdZe e zdze d<dZ e e d<dZedze d<dZeedze d <d Zee d <d Zee d <dZee d<fdZedZej2dZxZS)Sam3VisionConfiga fpn_hidden_size (`int`, *optional*, defaults to 256): The hidden dimension of the FPN. backbone_feature_sizes (`List[List[int]]`, *optional*, defaults to `[[288, 288], [144, 144], [72, 72]]`): The spatial sizes (height, width) of the feature maps from the backbone at different scales. scale_factors (`list[float]`, *optional*, defaults to `[4.0, 2.0, 1.0, 0.5]`): Scale factors for FPN multi-scale features. List of scaling factors for each FPN level. vision_configsam3_vision_modelrNfpn_hidden_sizebackbone_feature_sizes scale_factorsrrrrr(r)c |jgdn |j|_|jddgddgddgg|_t|jtrT|jj dd|jd<t |jddi|j|_n|jt d|_t| di|y)N)g@g@g?g?i Hr=rr/) rMrL isinstancerdictgetr r0r1r2s r6r1zSam3VisionConfig.__post_init__^s595G5G5O1UYUgUg  & & .,/:SzB8*LD ' d**D 1151E1E1I1I,Xh1iD  .#1$2F2F|2T#U#mX\XlXl#mD  ! ! )#12B#C#ED  ''r7c.|jjS)z"Image size for the vision encoder.rrr3s r6rzSam3VisionConfig.image_sizeks##...r7c&||j_y)z-Set the image size and propagate to backbone.NrUr3values r6rzSam3VisionConfig.image_sizeps+0'r7)r8r9r:r;r<r=r sub_configsrrRrr?rKr>rLr@rMrCrrBrr)r1propertyrsetterrDrEs@r6rGrGDs&O$J:K7;OT,,t3:OS*.D4K.(,M4;%,J NE #u# (//00r7rGceZdZUdZdZdZeed<dZeed<dZ eed<d Z eed <d Z e ezed <d Z eed<dZe ezed<dZe ed<dZeed<dZe ed<y)Sam3GeometryEncoderConfigzc roi_size (`int`, *optional*, defaults to 7): ROI size for box pooling operations. sam3_geometry_encoderrJrr num_layersrr皙?dropoutrelurrr'rrr+roi_sizer(r)N)r8r9r:r;r=rr>r?r`rrrdrCrrBr'rrfr)r/r7r6r^r^vs} )JKJ  !s!GUS[J"%NECK% NE Hc#u#r7r^ceZdZUdZdZdZeed<dZeed<dZ eed<d Z eed <d Z e ezed <d Z eed<dZe ezed<dZe ed<dZe ed<y)Sam3DETREncoderConfigzo hidden_dropout (`float`, *optional*, defaults to 0.0): Dropout probability for hidden states. sam3_detr_encoderrJrr`rarrbrrcrdrerrr'rrr(r)N)r8r9r:r;r=rr>r?r`rrrdrCrrBr'rr)r/r7r6rhrhss %JKJ  !s!GUS[J"%NECK% NE #u#r7rhceZdZUdZdZdZeed<dZeed<dZ eed<d Z eed <d Z eed <d Z e ezed<dZeed<dZe ezed<dZe ed<dZe ed<y)Sam3DETRDecoderConfigz] num_queries (`int`, *optional*, defaults to 200): Number of object queries. sam3_detr_decoderrJrrjr` num_queriesrarrbrrcrdrerrr'rrr(r)N)r8r9r:r;r=rr>r?r`rorrrdrCrrBr'rr)r/r7r6rlrls} %JKJK  !s!GUS[J"%NECK% NE #u#r7rlcpeZdZUdZdZdZeed<dZeed<dZ e ed<d Z e ezed <d Z eed <d Z e ed<y)Sam3MaskDecoderConfigz num_upsampling_stages (`int`, *optional*, defaults to 3): Number of upsampling stages in the pixel decoder (FPN). sam3_mask_decoderrJrrnum_upsampling_stagesrrrrdrarr(r)N)r8r9r:r;r=rr>r?rsrrCrdrr)r/r7r6rqrqsQ %JK!"3" NE GUS[  #u#r7rqceZdZUdZdZdZeeee e e dZ dZ eezdzed<dZeezdzed<dZeezdzed<dZeezdzed <dZeezdzed <dZeezdzed <d Zeed <fdZedZej6dZxZS) Sam3ConfigaD geometry_encoder_config (`dict` or `Sam3GeometryEncoderConfig`, *optional*): Configuration for the geometry encoder. detr_encoder_config (`dict` or `Sam3DETREncoderConfig`, *optional*): Configuration for the DETR encoder. detr_decoder_config (`dict` or `Sam3DETRDecoderConfig`, *optional*): Configuration for the DETR decoder. mask_decoder_config (`dict` or `Sam3MaskDecoderConfig`, *optional*): Configuration for the mask decoder. Example: ```python >>> from transformers import Sam3Config, Sam3Model >>> # Initializing a SAM3 configuration >>> configuration = Sam3Config() >>> # Initializing a model from the configuration >>> model = Sam3Model(configuration) >>> # Accessing the model configuration >>> configuration = model.config ``` sam3T)rH text_configgeometry_encoder_configdetr_encoder_configdetr_decoder_configmask_decoder_configNrHrwrxryrzr{r(r)c |jt|_t|jtrtd i|j|_|jt d iddddddddd |_t|jtrt d i|j|_|j t|_t|j trtd i|j |_|jt|_t|jtrtd i|j|_|jt|_ t|jtrtd i|j|_ |jt|_ t|jtrtd i|j|_ t|<d i|y) Niriir"rrr) vocab_sizerrprojection_dimrrmax_position_embeddingsrr/)rHrGrQrRrwrrxr^ryrhrzrlr{rqr0r1r2s r6r1zSam3Config.__post_init__s    %!1!3D  d(($ /!1!GD4F4F!GD     #- "'#')-&))++-/1"(  D  d&& --A0@0@AD   ' ' /+D+FD ( d22D 9+D+dtGcGc+dD (  # # +'<'>D $ d.. 5'<'Xt?W?W'XD $  # # +'<'>D $ d.. 5'<'Xt?W?W'XD $  # # +'<'>D $ d.. 5'<'Xt?W?W'XD $ ''r7c.|jjS)zImage size for the SAM3 model.rHrrVs r6rzSam3Config.image_size$s!!,,,r7c&||j_y)z2Set the image size and propagate to vision config.NrrXs r6rzSam3Config.image_size)s).%r7)r8r9r:r;r=is_compositionrGrr^rhrlrqrZrHrRrr?rwrxryrzr{r)rCr1r[rr\rDrEs@r6rurus2JN)%#<444 K59M4**T1826K((4/6>BT$44t;B:> 0047>:> 0047>:> 0047>#u#*(X--..r7ru)rur rGr^rhrlrqN)r;huggingface_hub.dataclassesr transformersrconfiguration_utilsrutilsrautor r r rGr^rhrlrqru__all__r/r7r6rsE.'3#-?+&7$&7,&7R?+-0'-0,-0`?+$ 0$,$(?+$,$,$&?+$,$,$(?+ $, $, $ ?+a.!a.,a.H r7