a ¾ ÀhÅã@sLddlmZddlmZddlmZmZe e¡Z Gdd„deƒZ dgZ dS)é)ÚPretrainedConfig)Úloggingé)ÚCONFIG_MAPPINGÚ AutoConfigcsJeZdZdZdZddddœZeedœZdeeeee dœ‡fdd„ Z ‡Z S)ÚShieldGemma2Configaá This is the configuration class to store the configuration of a [`ShieldGemma2ForImageClassification`]. It is used to instantiate an ShieldGemma2ForImageClassification according to the specified arguments, defining the model architecture. Instantiating a configuration with the defaults will yield a similar configuration to that of the shieldgemma-2-4b-it. e.g. [google/gemma-3-4b](https://huggingface.co/google/gemma-3-4b) 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 (`Union[ShieldGemma2TextConfig, dict]`, *optional*): The config object of the text backbone. vision_config (`Union[AutoConfig, dict]`, *optional*): Custom vision config or dict. mm_tokens_per_image (`int`, *optional*, defaults to 256): The number of tokens per image embedding. boi_token_index (`int`, *optional*, defaults to 255999): The begin-of-image token index to wrap the image prompt. eoi_token_index (`int`, *optional*, defaults to 256000): The end-of-image token index to wrap the image prompt. image_token_index (`int`, *optional*, defaults to 262144): The image token index to encode the image prompt. initializer_range (`float`, *optional*, defaults to 0.02): The standard deviation of the truncated_normal_initializer for initializing all weight matrices. Example: ```python >>> from transformers import ShieldGemma2ForConditionalGeneration, ShieldGemma2Config, SiglipVisionConfig, ShieldGemma2TextConfig >>> # Initializing a Siglip-like vision config >>> vision_config = SiglipVisionConfig() >>> # Initializing a ShieldGemma2 Text config >>> text_config = ShieldGemma2TextConfig() >>> # Initializing a ShieldGemma2 gemma-3-4b style configuration >>> configuration = ShieldGemma2Config(vision_config, text_config) >>> # Initializing a model from the gemma-3-4b style configuration >>> model = ShieldGemma2TextConfig(configuration) >>> # Accessing the model configuration >>> configuration = model.config ```Z shieldgemma2Úimage_token_indexÚboi_token_indexÚeoi_token_index)Zimage_token_idZ boi_token_idZ eoi_token_id)Ú text_configÚ vision_configNééÿçéèéç{®Gáz”?)Úmm_tokens_per_imager r rÚinitializer_rangec  sÎt|tƒr2| dd¡|d<t|dfi|¤Ž}n|durDtdƒ}||_t|tƒr|| dd¡|d<t|dfi|¤Ž}n|durŽtdƒ}||_||_||_||_||_||_ ||_ t ƒj fi|¤ŽdS)NÚ model_typeZsiglip_vision_modelZ gemma3_text) Ú isinstanceÚdictÚgetrr r rr r rrÚsuperÚ__init__) Úselfr r rr r rrÚkwargs©Ú __class__©úw/var/www/html/assistant/venv/lib/python3.9/site-packages/transformers/models/shieldgemma2/configuration_shieldgemma2.pyrRs&    zShieldGemma2Config.__init__)NNr rrrr) Ú__name__Ú __module__Ú __qualname__Ú__doc__rZ attribute_maprZ sub_configsÚintÚfloatrÚ __classcell__rrrrrs*0ý øørN) Zconfiguration_utilsrÚutilsrÚautorrZ get_loggerr ÚloggerrÚ__all__rrrrÚs   ]