`i|dZddlZddlmZddlmZddlmZmZej e Z Gdd eZ d gZ dS) zColPali model configurationN)deepcopy)PretrainedConfig)CONFIG_MAPPING AutoConfigc>eZdZdZdZeedZ ddeffd Z xZ S) ColPaliConfigab Configuration class to store the configuration of a [`ColPaliForRetrieval`]. It is used to instantiate an instance of `ColPaliForRetrieval` according to the specified arguments, defining the model architecture following the methodology from the "ColPali: Efficient Document Retrieval with Vision Language Models" paper. Creating a configuration with the default settings will result in a configuration where the VLM backbone is set to the default PaliGemma configuration, i.e the one from [vidore/colpali-v1.2](https://huggingface.co/vidore/colpali-v1.2). Note that contrarily to what the class name suggests (actually the name refers to the ColPali **methodology**), you can use a different VLM backbone model than PaliGemma by passing the corresponding VLM configuration to the class constructor. Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the documentation from [`PretrainedConfig`] for more information. Args: vlm_config (`PretrainedConfig`, *optional*): Configuration of the VLM backbone model. text_config (`PretrainedConfig`, *optional*): Configuration of the text backbone model. Overrides the `text_config` attribute of the `vlm_config` if provided. embedding_dim (`int`, *optional*, defaults to 128): Dimension of the multi-vector embeddings produced by the model. Example: ```python from transformers.models.colpali import ColPaliConfig, ColPaliForRetrieval config = ColPaliConfig() model = ColPaliForRetrieval(config) ``` colpali) vlm_config text_configN embedding_dimc |/td}tdnt|trct |}d|vrt d|dtvrtd|ddt|dd i|}n5t|ts tdt|d||_ ||n|j |_ t|j tr6| dd |d<t|dd i||_ ||_tjd i|dS) N paligemmaza`vlm_config` is `None`. Initializing `vlm_config` with the `PaliGemmaConfig` with default values. model_typez^The `model_type` key is missing in the `vlm_config` dictionary. Please provide the model type.zThe model type `z6` is not supported. Please provide a valid model type.zWInvalid type for `vlm_config`. Expected `PretrainedConfig`, `dict`, or `None`, but got .gemma)rloggerinfo isinstancedictrKeyError ValueErrorr TypeErrortyper r getrsuper__init__)selfr r rkwargs __class__s /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/transformers/models/colpali/configuration_colpali.pyr zColPaliConfig.__init__?s  ' 466J KKs     D ) ) !*--J:--tL)?? wz,'?www( <(@AOOJOOJJJ(899 }jnoyjzjz}}} %*5*A;;zG] d& - - X(3 g(N(NK %-k,.GHWW;WWD *""6""""")NNr) __name__ __module__ __qualname____doc__rrr sub_configsintr __classcell__)r#s@r$r r sv@J!1*MMK $#$# $#$#$#$#$#$#$#$#$#$#r%r )r)loggingcopyrconfiguration_utilsrautorr getLoggerr&rr __all__rr%r$r3s"!333333--------  8 $ $H#H#H#H#H#$H#H#H#V  r%