a h@sXdZddlZddlmZddlmZddlmZmZe e Z Gdd d eZ d gZ dS) zColPali model configurationN)deepcopy)PretrainedConfig)CONFIG_MAPPING AutoConfigcs6eZdZdZdZeedZd edfdd Z Z 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) ``` Zcolpali) vlm_config text_configN) embedding_dimc s|durtd}tdnt|trzt|}d|vrBtdn |dtvrbtd|ddt|dfi|}n$t|tr|}nt dt |d||_ |dur|n|j |_ t|j tr| dd |d<t|dfi||_ ||_tjfi|dS) NZ 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 .Zgemma)rloggerinfo isinstancedictrKeyError ValueErrorr TypeErrortyper r getr super__init__)selfr r r kwargs __class__m/var/www/html/assistant/venv/lib/python3.9/site-packages/transformers/models/colpali/configuration_colpali.pyr?s8     zColPaliConfig.__init__)NNr ) __name__ __module__ __qualname____doc__r rrZ sub_configsintr __classcell__rrrrrs  r)r#loggingcopyrZconfiguration_utilsrautorr getLoggerr rr__all__rrrrs   M