S^(hxdZddlZddlmZddlmZddlmZmZeje Z Gdd eZ d gZ y) zColPali model configurationN)deepcopy)PretrainedConfig)CONFIG_MAPPING AutoConfigc>eZdZdZdZeedZ ddeffd Z xZ S) ColPaliConfiga 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). The ColPali config is very similar to [`PaligemmaConfig`], but with an extra attribute defining the embedding dimension. 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_config embedding_dimc <|#td}tjdnt|trIt |}d|vr t d|dtvrtd|ddt|dd i|}n+t|tr|}ntdt|d||_ ||n |jx|_ }t|jtr$d|vr|dnd |d<t|dd i||_ ||_ t|<d i|y) 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 rsuper__init__)selfr r rkwargs __class__s o/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/transformers/models/colpali/configuration_colpali.pyrzColPaliConfig.__init__AsO  ' 46J KKs  D )!*-J:-tL)? &z,'?&@@vw( <(@AOJOJ  $4 5#Jijnoyjzi{{|} %8C8OU_UkUkk; d&& -EQU`E` L(AfmK %-k,.GHW;WD * "6")NN) __name__ __module__ __qualname____doc__rrr sub_configsintr __classcell__)r!s@r"r r s9 DJ!1*MK &# &#&#r#r )r(loggingcopyrconfiguration_utilsrautorr getLoggerr%rr __all__rr#r"r2sD"3-   8 $L#$L#^  r#