gP dZddlZddlmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZer dd lmZdd lmZejeZGd d e ZGd de ZGdde ZGdde ZdS)zGroupViT model configurationN OrderedDict) TYPE_CHECKINGAnyMappingOptionalUnion)PretrainedConfig) OnnxConfig)logging)ProcessorMixin) TensorTypeceZdZdZdZ dfd Zedeee j fddfdZ xZ S)GroupViTTextConfiga> This is the configuration class to store the configuration of a [`GroupViTTextModel`]. It is used to instantiate an GroupViT model according to the specified arguments, defining the model architecture. Instantiating a configuration with the defaults will yield a similar configuration to that of the GroupViT [nvidia/groupvit-gcc-yfcc](https://huggingface.co/nvidia/groupvit-gcc-yfcc) architecture. Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the documentation from [`PretrainedConfig`] for more information. Args: vocab_size (`int`, *optional*, defaults to 49408): Vocabulary size of the GroupViT text model. Defines the number of different tokens that can be represented by the `inputs_ids` passed when calling [`GroupViTModel`]. hidden_size (`int`, *optional*, defaults to 256): Dimensionality of the encoder layers and the pooler layer. intermediate_size (`int`, *optional*, defaults to 1024): Dimensionality of the "intermediate" (i.e., feed-forward) layer in the Transformer encoder. num_hidden_layers (`int`, *optional*, defaults to 12): Number of hidden layers in the Transformer encoder. num_attention_heads (`int`, *optional*, defaults to 4): Number of attention heads for each attention layer in the Transformer encoder. max_position_embeddings (`int`, *optional*, defaults to 77): The maximum sequence length that this model might ever be used with. Typically set this to something large just in case (e.g., 512 or 1024 or 2048). hidden_act (`str` or `function`, *optional*, defaults to `"quick_gelu"`): The non-linear activation function (function or string) in the encoder and pooler. If string, `"gelu"`, `"relu"`, `"selu"` and `"gelu_new"` `"quick_gelu"` are supported. layer_norm_eps (`float`, *optional*, defaults to 1e-5): The epsilon used by the layer normalization layers. attention_dropout (`float`, *optional*, defaults to 0.0): The dropout ratio for the attention probabilities. dropout (`float`, *optional*, defaults to 0.0): The dropout probability for all fully connected layers in the embeddings, encoder, and pooler. initializer_range (`float`, *optional*, defaults to 0.02): The standard deviation of the truncated_normal_initializer for initializing all weight matrices. initializer_factor (`float`, *optional*, defaults to 1.0): A factor for initializing all weight matrices (should be kept to 1, used internally for initialization testing). Example: ```python >>> from transformers import GroupViTTextConfig, GroupViTTextModel >>> # Initializing a GroupViTTextModel with nvidia/groupvit-gcc-yfcc style configuration >>> configuration = GroupViTTextConfig() >>> model = GroupViTTextModel(configuration) >>> # Accessing the model configuration >>> configuration = model.config ```groupvit_text_model M quick_geluh㈵>{Gz??c tjd| ||d|||_||_||_| |_||_||_||_||_ ||_ | |_ | |_ | |_ dS)N) pad_token_id bos_token_id eos_token_id)super__init__ vocab_size hidden_sizeintermediate_sizedropoutnum_hidden_layersnum_attention_headsmax_position_embeddingslayer_norm_eps hidden_actinitializer_rangeinitializer_factorattention_dropout)selfr(r)r*r,r-r.r0r/r+r3r1r2r"r#r$kwargs __class__s o/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/models/groupvit/configuration_groupvit.pyr'zGroupViTTextConfig.__init__Zs& sl\hsslrsss$&!2 !2#6 '>$,$!2"4!2pretrained_model_name_or_pathreturnr c N|||j|fi|\}}|ddkr|d}d|vrMt|dr=|d|jkr,t d|dd|jd|j|fi|S)N model_typegroupvit text_configYou are using a model of type to instantiate a model of type N. This is not supported for all configurations of models and can yield errors._set_token_in_kwargsget_config_dictgethasattrr<loggerwarning from_dictclsr9r5 config_dicts r7from_pretrainedz"GroupViTTextConfig.from_pretrained|s   (((1c12OZZSYZZ V ??< ( (J 6 6%m4K ; & &73 +E+E &+VbJcgjguJuJu NNr\1Jrr>rrr    s}[33F333r8)rrrrrrrrrrrrrrr __name__ __module__ __qualname____doc__r<r' classmethodr strosPathLikerM __classcell__r6s@r7rr"s3 3 j'J "! 3 3 3 3 3 3D4E#r{BR>> from transformers import GroupViTVisionConfig, GroupViTVisionModel >>> # Initializing a GroupViTVisionModel with nvidia/groupvit-gcc-yfcc style configuration >>> configuration = GroupViTVisionConfig() >>> model = GroupViTVisionModel(configuration) >>> # Accessing the model configuration >>> configuration = model.config ```groupvit_vision_modelii)r r r)@r)r]r^r^r\r gelurrrrg?rc tjdi|||_||_||_|t |kr-t d|dt |||_||_ ||_ ||_ ||_ | |_ | |_| |_| |_| |_||_||_||_||_||_dS)Nz&Manually setting num_hidden_layers to z1, but we expect num_hidden_layers = sum(depth) = r%)r&r'r)r*depthssumrGrHr,num_group_tokensnum_output_groupsr- image_size patch_size num_channelsr0r/r+r3r1r2 assign_epsassign_mlp_ratio)r4r)r*rcr,rerfr-rgrhrir0r/r+r3r1r2rjrkr5r6s r7r'zGroupViTVisionConfig.__init__s, ""6"""&!2 F + + NN/9J//!$V//   "3 0!2#6 $$($, !2!2"4$ 0r8r9r:r c N|||j|fi|\}}|ddkr|d}d|vrMt|dr=|d|jkr,t d|dd|jd|j|fi|S)Nr<r= vision_configr?r@rArBrJs r7rMz$GroupViTVisionConfig.from_pretraineds   (((1c12OZZSYZZ V ??< ( (J 6 6%o6K ; & &73 +E+E &+VbJcgjguJuJu NNr\1Jrr>rrr    s}[33F333r8rNrXs@r7rZrZs5 5 n)Jyy#$**q'.1.1.1.1.1.1`4E#r{BR| || kr2| dvr.| |vr d| d| d} nd| d} t | H|| ||i}tdi|} d | vr'd | d D| d <| D]G\} } | |vr>| || kr2| dvr.| |vr d| d | d} nd | d} t | H|| |i}t d |i}t dtdi||_ tdi||_ ||_ ||_ ||_d|_d|_d|_dS)Ntext_config_dictvision_config_dict)transformers_version`zp` is found in both `text_config_dict` and `text_config` but with different values. The value `text_config_dict["z"]` will be used instead.zn`text_config_dict` is provided which will be used to initialize `GroupViTTextConfig`. The value `text_config["z"]` will be overridden.id2labelc4i|]\}}t||Sr%)rT).0keyvalues r7 z+GroupViTConfig.__init__.._s1333(2UCHHe333r8zv` is found in both `vision_config_dict` and `vision_config` but with different values. The value `vision_config_dict["zt`vision_config_dict` is provided which will be used to initialize `GroupViTVisionConfig`. The value `vision_config["zS`text_config` is `None`. Initializing the `GroupViTTextConfig` with default values.zW`vision_config` is `None`. initializing the `GroupViTVisionConfig` with default values.rrFr%)popr&r'rto_dictitemsrGinfoupdaterZr>rmprojection_dimprojection_intermediate_dimlogit_scale_init_valuer1r2output_segmentation)r4r>rmrrrr5rsrt_text_config_dictrzr{message_vision_config_dictr6s r7r'zGroupViTConfig.__init__(s1"::&8$??#ZZ(rmc `|d||d|S)z Instantiate a [`GroupViTConfig`] (or a derived class) from groupvit text model configuration and groupvit vision model configuration. Returns: [`GroupViTConfig`]: An instance of a configuration object )r>rmr%)r~)rKr>rmr5s r7from_text_vision_configsz'GroupViTConfig.from_text_vision_configss:sf{2244MDYDYD[D[ff_efffr8)NNrrprq) rOrPrQrRr<r'rSrrZrrWrXs@r7roro s2J$(% _)_)_)_)_)_)B g3E gVj g g g[ g g g g gr8roc eZdZedeeeeefffdZedeeeeefffdZede fdZ dddd ed ed e d deee ff fd Z edefdZxZS)GroupViTOnnxConfigr:c Htddddfdddddd fd dddfgS) N input_idsbatchsequence)rr pixel_valuesriheightwidth)rrr attention_maskrr4s r7inputszGroupViTOnnxConfig.inputssN'j99:WHQX!Y!YZ!w:#>#>?    r8cHtdddifdddifdddifdddifgS)Nlogits_per_imagerrlogits_per_text text_embeds image_embedsrrs r7outputszGroupViTOnnxConfig.outputssI#a\2"QL1G -!W.     r8cdS)Ng-C6?r%rs r7atol_for_validationz&GroupViTOnnxConfig.atol_for_validationstr8N processorr batch_size seq_length frameworkrct|j|||}t|j||}i||S)N)rrr)rr)r&generate_dummy_inputs tokenizerimage_processor)r4rrrrtext_input_dictimage_input_dictr6s r7rz(GroupViTOnnxConfig.generate_dummy_inputssn ''77  J:Yb8  !7788  %* 9  7/6%566r8cdS)Nr%rs r7default_onnx_opsetz%GroupViTOnnxConfig.default_onnx_opsetsrr8)rrN)rOrPrQpropertyrrTintrrfloatrrrrrrWrXs@r7rrsK  WS#X%6 67   X  gc3h&7!78   X UX ,0 7 7# 7 7 7 L) 7 c  7 7 7 7 7 7CXr8r)rRrU collectionsrtypingrrrrr configuration_utilsr onnxr utilsr processing_utilsrr get_loggerrOrGrrZrorr%r8r7rs#" ######??????????????333333$222222######  H % %j4j4j4j4j4)j4j4j4Zz4z4z4z4z4+z4z4z4zGgGgGgGgGg%GgGgGgT++++++++++r8