o Zh%@sLddlmZddlmZddlmZmZeeZ GdddeZ dgZ dS))PretrainedConfig)logging)CONFIG_MAPPING AutoConfigc s`eZdZdZdZddddZeedZ   ddedededede f 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 shieldgemma2image_token_indexboi_token_indexeoi_token_index)Zimage_token_idZ boi_token_idZ eoi_token_id) text_config vision_configN{Gz?mm_tokens_per_imageinitializer_rangec  st|trd|vr |dnd|d<t|ddi|}n |dur&td}||_t|trFd|vr6|dnd|d<t|ddi|}n |durOtd}||_||_||_||_||_||_||_ t j di|dS)N model_typeZsiglip_vision_modelZ gemma3_text) isinstancedictrr r rr r rrsuper__init__) selfr r rr r rrkwargs __class__rj/var/www/auris/lib/python3.10/site-packages/transformers/models/shieldgemma2/configuration_shieldgemma2.pyrRs(    zShieldGemma2Config.__init__)NNr rrrr) __name__ __module__ __qualname____doc__rZ attribute_maprZ sub_configsintfloatr __classcell__rrrrrs40 rN) Zconfiguration_utilsrutilsrautorrZ get_loggerrloggerr__all__rrrrs    _