
    i                         d Z ddlmZ ddlmZ ddlmZmZ ddlm	Z	m
Z
  ej                  e      Z ed	      e G d
 de                    ZdgZy)zColPali model configuration    )strict   )PreTrainedConfig)auto_docstringlogging   )CONFIG_MAPPING
AutoConfigzvidore/colpali-v1.2)
checkpointc                   t     e Zd ZU dZdZeedZdZe	ez  dz  e
d<   dZe	ez  dz  e
d<   dZee
d<    fd	Z xZS )
ColPaliConfigz
    Example:

    ```python
    from transformers.models.colpali import ColPaliConfig, ColPaliForRetrieval

    config = ColPaliConfig()
    model = ColPaliForRetrieval(config)
    ```
    colpali)
vlm_configtext_configNr   r      embedding_dimc                 T   | j                   (t        d          | _         t        j                  d       nDt	        | j                   t
              r*t        | j                   d      di | j                   | _         | j                  | j                  n| j                   j                  | _        t	        | j                  t
              rS| j                  j                  dd      | j                  d<   t        | j                  d      di | j                  | _        t        | $  di | y )N	paligemmaza`vlm_config` is `None`. Initializing `vlm_config` with the `PaliGemmaConfig` with default values.
model_typegemma )
r   r	   loggerinfo
isinstancedictr   getsuper__post_init__)selfkwargs	__class__s     /var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/colpali/configuration_colpali.pyr   zColPaliConfig.__post_init__/   s    ??",[9;DOKKs .,T__\-JK^doo^DO/3/?/?/K4++QUQ`Q`QlQld&&--1-=-=-A-A,PW-XD\*-d.>.>|.LMaPTP`P`aD''    )__name__
__module____qualname____doc__r   r   r
   sub_configsr   r   __annotations__r   r   intr   __classcell__)r!   s   @r"   r   r      sW    	 J!1*MK15J''$.526K((4/6M3( (r#   r   N)r'   huggingface_hub.dataclassesr   configuration_utilsr   utilsr   r   autor	   r
   
get_loggerr$   r   r   __all__r   r#   r"   <module>r2      s_    " . 3 , - 
		H	% 01!($ !(  2!(H 
r#   