U ,-exQ@sdZddlZddlmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZertdd lmZdd lmZeeZd d iZGd dde ZGddde ZGddde ZGddde ZdS)z GroupViT model configurationN OrderedDict) TYPE_CHECKINGAnyMappingOptionalUnion)PretrainedConfig) OnnxConfig)logging)ProcessorMixin) TensorTypeznvidia/groupvit-gcc-yfcczHhttps://huggingface.co/nvidia/groupvit-gcc-yfcc/resolve/main/config.jsoncsDeZdZdZdZdfdd Zeeee j fddddZ Z 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 ```Zgroupvit_text_model M quick_geluh㈵>{Gz??c sftjf| ||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)r+r*r&r.r,r-rrr kwargs __class__t/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/transformers/models/groupvit/configuration_groupvit.pyr"]szGroupViTTextConfig.__init__r pretrained_model_name_or_pathreturncKs~|||j|f|\}}|ddkr2|d}d|krpt|drp|d|jkrptd|dd|jd|j|f|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.Z_set_token_in_kwargsZget_config_dictgethasattrr8loggerwarning from_dictclsr6r0Z config_dictr3r3r4from_pretraineds  z"GroupViTTextConfig.from_pretrained)rrrrrrrrrrrrrrr __name__ __module__ __qualname____doc__r8r" classmethodrstrosPathLikerF __classcell__r3r3r1r4r&s(4"rcs~eZdZdZdZdddddgddd d gdd d gdd d dd ddddddddgffdd Zeeee j fddddZ Z S)GroupViTVisionConfigaB This is the configuration class to store the configuration of a [`GroupViTVisionModel`]. 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: hidden_size (`int`, *optional*, defaults to 384): Dimensionality of the encoder layers and the pooler layer. intermediate_size (`int`, *optional*, defaults to 1536): Dimensionality of the "intermediate" (i.e., feed-forward) layer in the Transformer encoder. depths (`List[int]`, *optional*, defaults to [6, 3, 3]): The number of layers in each encoder block. num_group_tokens (`List[int]`, *optional*, defaults to [64, 8, 0]): The number of group tokens for each stage. num_output_groups (`List[int]`, *optional*, defaults to [64, 8, 8]): The number of output groups for each stage, 0 means no group. num_attention_heads (`int`, *optional*, defaults to 6): Number of attention heads for each attention layer in the Transformer encoder. image_size (`int`, *optional*, defaults to 224): The size (resolution) of each image. patch_size (`int`, *optional*, defaults to 16): The size (resolution) of each patch. hidden_act (`str` or `function`, *optional*, defaults to `"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. dropout (`float`, *optional*, defaults to 0.0): The dropout probabilitiy for all fully connected layers in the embeddings, encoder, and pooler. attention_dropout (`float`, *optional*, defaults to 0.0): The dropout ratio for the attention probabilities. 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 GroupViTVisionConfig, GroupViTVisionModel >>> # Initializing a GroupViTVisionModel with nvidia/groupvit-gcc-yfcc style configuration >>> configuration = GroupViTVisionConfig() >>> model = GroupViTVisionModel(configuration) >>> # Accessing the model configuration >>> configuration = model.config ```Zgroupvit_vision_modeliir r@rZgelurrrrg?rc stjf|||_||_||_|t|krFtd|dt|||_||_ ||_ ||_ ||_ | |_ | |_| |_| |_| |_||_||_||_||_||_dS)Nz&Manually setting num_hidden_layers to z1, but we expect num_hidden_layers = sum(depth) = )r!r"r$r%depthssumrArBr'num_group_tokensnum_output_groupsr( image_size patch_size num_channelsr+r*r&r.r,r- assign_epsassign_mlp_ratio)r/r$r%rWr'rYrZr(r[r\r]r+r*r&r.r,r-r^r_r0r1r3r4r"s. zGroupViTVisionConfig.__init__r r5cKs~|||j|f|\}}|ddkr2|d}d|krpt|drp|d|jkrptd|dd|jd|j|f|S)Nr8r9 vision_configr;r<r=r>rDr3r3r4rFs  z$GroupViTVisionConfig.from_pretrainedrGr3r3r1r4rQs.70rQcs:eZdZdZdZd fdd Zeeed d d Z Z S) GroupViTConfiga [`GroupViTConfig`] is the configuration class to store the configuration of a [`GroupViTModel`]. It is used to instantiate a GroupViT model according to the specified arguments, defining the text model and vision model configs. 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: text_config (`dict`, *optional*): Dictionary of configuration options used to initialize [`GroupViTTextConfig`]. vision_config (`dict`, *optional*): Dictionary of configuration options used to initialize [`GroupViTVisionConfig`]. projection_dim (`int`, *optional*, defaults to 256): Dimentionality of text and vision projection layers. projection_intermediate_dim (`int`, *optional*, defaults to 4096): Dimentionality of intermediate layer of text and vision projection layers. logit_scale_init_value (`float`, *optional*, defaults to 2.6592): The inital value of the *logit_scale* parameter. Default is used as per the original GroupViT implementation. kwargs (*optional*): Dictionary of keyword arguments. r9Nr/L F@c s|dd}|dd}tjf||dk r|dkr:i}tf|} | D]V\} } | |krP| || krP| dkrP| |krd| d| d} n d| d} t| qP|| |dk rl|dkri}t f|} d | krd d | d D| d <| D]`\} } | |kr| || kr| dkr| |krHd| d | d} n d | d} t| q|| |dkri}t d|dkri}t dtf||_ t f||_ ||_ ||_||_d|_d|_d|_dS)Ntext_config_dictvision_config_dict)Ztransformers_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 overriden.Zid2labelcSsi|]\}}t||qSr3)rM).0keyvaluer3r3r4 bsz+GroupViTConfig.__init__..zv` 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.rrF)popr!r"rto_dictitemsrArBupdaterQinfor:r`projection_dimprojection_intermediate_dimlogit_scale_init_valuer,r-Zoutput_segmentation)r/r:r`rprqrrr0rdreZ_text_config_dictrhrimessageZ_vision_config_dictr1r3r4r"+s^       "         zGroupViTConfig.__init__r:r`cKs|f||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 rt)rl)rEr:r`r0r3r3r4from_text_vision_configss z'GroupViTConfig.from_text_vision_configs)NNrrbrc) rHrIrJrKr8r"rLrrQrurPr3r3r1r4rasaracseZdZeeeeeeffdddZeeeeeeffdddZee dddZ dd eee d eee fd fd d Z eedddZZS)GroupViTOnnxConfig)r7cCs0tddddfdddddd fd dddfgS) NZ input_idsbatchsequence)rrZ pixel_valuesr]heightwidth)rrr Zattention_maskrr/r3r3r4inputss   zGroupViTOnnxConfig.inputscCs0tdddifdddifdddifdddifgS)NZlogits_per_imagerrwZlogits_per_textZ text_embedsZ image_embedsrr|r3r3r4outputss    zGroupViTOnnxConfig.outputscCsdS)Ng-C6?r3r|r3r3r4atol_for_validationsz&GroupViTOnnxConfig.atol_for_validationNr r) processor batch_size seq_length frameworkr7cs2tj|j|||d}tj|j||d}||S)N)rrr)rr)r!generate_dummy_inputsZ tokenizerZimage_processor)r/rrrrZtext_input_dictZimage_input_dictr1r3r4rsz(GroupViTOnnxConfig.generate_dummy_inputscCsdS)Nr3r|r3r3r4default_onnx_opsetsz%GroupViTOnnxConfig.default_onnx_opset)rrN)rHrIrJpropertyrrMintr}r~floatrrrrrrPr3r3r1r4rvs$   rv)rKrN collectionsrtypingrrrrrZconfiguration_utilsr Zonnxr utilsr Zprocessing_utilsr rZ get_loggerrHrAZ&GROUPVIT_PRETRAINED_CONFIG_ARCHIVE_MAPrrQrarvr3r3r3r4s$       l}