tjy*dZddlmZddlmZddlmZddlmZddl m Z m Z ed eGd d eZ ed eGd deZ ed eGddeZed eGddeZed eGddeZed eGddeZed eGddeZgdZdS)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 ^tjdi||j gd|_dSdS)N))super __post_init__r$selfkwargs __class__s i/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/transformers/models/sam3/configuration_sam3.pyr1zSam3ViTConfig.__post_init__>sC'''''  # +'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/188846Jd3i%S/1666J NE   %(us{(((JK,0cT)000+/EDL///=@tCy5c?:@@@"%NECK%%%#u###777777777r7r ceZdZUdZdZdZdeiZdZe e zdze d<dZ e e d<dZedze d<dZeedze d <d Zee d <d Zee d <dZee d<fdZedZejdZxZS)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|jtrK|jdd|jd<t |jddi|j|_n |jt d|_tjdi|dS)N)g@g@g?g?i Hr=rr/) rMrL isinstancerdictgetr r0r1r2s r6r1zSam3VisionConfig.__post_init__^s595G5O1111UYUg  & .,/:SzB8*LD ' d*D 1 1 F151E1I1I,Xh1i1iD  .#1$2F|2T#U#m#mX\Xl#m#mD  ! )#12B#C#E#ED '''''''r7c|jjS)z"Image size for the vision encoder.rrr3s r6rzSam3VisionConfig.image_sizeks#..r7c||j_dS)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### ( ( ( ( (//X/0000000r7rGceZdZUdZdZdZeed<dZeed<dZ eed<d Z eed <d Z e ezed <d Z eed<dZe ezed<dZe ed<dZeed<dZe ed<dS)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<dS)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/r7r6rhrhs %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<dS)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#####r7rlcreZdZUdZdZdZeed<dZeed<dZ e ed<d Z e ezed <d Z eed <d Z e ed<dS)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/r7r6rqrqs %JK!"3""" NE   GUS[    #u#####r7rqceZdZUdZdZdZeeee e e dZ dZ eezdzed<dZeezdzed<dZeezdzed<dZeezdzed <dZeezdzed <dZeezdzed <d Zeed <fdZedZejdZxZS) 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|_|jt|_t|jtrtd i|j|_|jt|_t|jtrtd i|j|_|j t|_ t|j trtd i|j |_ |j t|_ t|j trtd i|j |_ tjd i|dS) Niriir"rrr) vocab_sizerrprojection_dimrrmax_position_embeddingsrr/)rHrGrQrRrwrrxr^ryrhrzrlr{rqr0r1r2s r6r1zSam3Config.__post_init__s   %!1!3!3D  d($ / / H!1!G!GD4F!G!GD    #-  "'#')-&))++-/1"(    D  d& - - B-AA0@AAD   ' /+D+F+FD ( d2D 9 9 e+D+d+dtGc+d+dD (  # +'<'>'>D $ d. 5 5 Y'<'X'Xt?W'X'XD $  # +'<'>'>D $ d. 5 5 Y'<'X'Xt?W'X'XD $  # +'<'>'>D $ d. 5 5 Y'<'X'Xt?W'X'XD $'''''''r7c|jjS)zImage size for the SAM3 model.rHrrVs r6rzSam3Config.image_size$s!,,r7c||j_dS)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@r6rurusb2JN)%#<444 K59M4**T188826K((4/666>BT$44t;BBB:> 0047>>>:> 0047>>>:> 0047>>>#u###*(*(*(*(*(X--X-.......r7ru)rur rGr^rhrlrqN)r;huggingface_hub.dataclassesr transformersrconfiguration_utilsrutilsrautor r r rGr^rhrlrqru__all__r/r7r6rs......''''''333333######--------?+++&7&7&7&7&7$&7&7,+&7R?+++-0-0-0-0-0'-0-0,+-0`?+++$$$$$ 0$$,+$(?+++$$$$$,$$,+$&?+++$$$$$,$$,+$(?+++ $ $ $ $ $, $ $,+ $ ?+++a.a.a.a.a.!a.a.,+a.H   r7