rhlddlmZddlmZddlmZmZejeZ GddeZ dgZ y))PretrainedConfig)logging)CONFIG_MAPPING AutoConfigc beZdZdZdZddddZeedZ d dedededed e f fd Z xZ 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 ``` shieldgemma2image_token_indexboi_token_indexeoi_token_index)image_token_id boi_token_id eoi_token_id) text_config vision_configmm_tokens_per_imageinitializer_rangec t|tr'|jdd|d<t|ddi|}n| td}||_t|tr'|jdd|d<t|ddi|}n| td}||_||_||_||_||_||_ ||_ t |0di|y)N model_typesiglip_vision_model gemma3_text) isinstancedictgetrrrrr r r rsuper__init__) selfrrrr r r rkwargs __class__s /var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/transformers/models/shieldgemma2/configuration_shieldgemma2.pyrzShieldGemma2Config.__init__Rs mT **7*;*;LJ_*`M, '*=+FGX-XM  "*+@ACM* k4 ((3 m(TK %(\)BCRkRK  (79K&*#6 ..!2!2 "6")NNiiig{Gz?) __name__ __module__ __qualname____doc__r attribute_mapr sub_configsintfloatr __classcell__)r!s@r"r r s. ` J-))M #-zJK#&&&!(#'!#! !#  !#  !#!#!!#!#r#r N) configuration_utilsrutilsrautorr get_loggerr%loggerr __all__rr#r"r4s>"4-   H %Z#)Z#z r#