
    sg                     n    d Z ddlZddlmZmZmZ ddlmZ ddlm	Z	m
Z
mZmZmZ ddlmZ  G d d	e      Zy)
z
Processor class for ViLT.
    N)ListOptionalUnion   )ProcessorMixin)BatchEncodingPaddingStrategyPreTokenizedInput	TextInputTruncationStrategy)
TensorTypec            "       2    e Zd ZdZddgZdZdZd fd	Z	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ddee	e
ee	   ee
   f   ded	eeeef   d
eeeef   dee   dedee   dee   dee   dedededededeeeef      def dZd Zd Zed        Zed        Zed        Z xZS )ViltProcessorau  
    Constructs a ViLT processor which wraps a BERT tokenizer and ViLT image processor into a single processor.

    [`ViltProcessor`] offers all the functionalities of [`ViltImageProcessor`] and [`BertTokenizerFast`]. See the
    docstring of [`~ViltProcessor.__call__`] and [`~ViltProcessor.decode`] for more information.

    Args:
        image_processor (`ViltImageProcessor`, *optional*):
            An instance of [`ViltImageProcessor`]. The image processor is a required input.
        tokenizer (`BertTokenizerFast`, *optional*):
            An instance of ['BertTokenizerFast`]. The tokenizer is a required input.
    image_processor	tokenizerViltImageProcessor)BertTokenizerBertTokenizerFastc                     d }d|v r+t        j                  dt               |j                  d      }||n|}|t	        d      |t	        d      t
        |   ||       | j                  | _        y )Nfeature_extractorzhThe `feature_extractor` argument is deprecated and will be removed in v5, use `image_processor` instead.z)You need to specify an `image_processor`.z"You need to specify a `tokenizer`.)	warningswarnFutureWarningpop
ValueErrorsuper__init__r   current_processor)selfr   r   kwargsr   	__class__s        [/var/www/html/venv/lib/python3.12/site-packages/transformers/models/vilt/processing_vilt.pyr   zViltProcessor.__init__-   s     &(MM
 !'

+> ?-<-H/N_"HIIABB)4!%!5!5    textadd_special_tokenspadding
truncation
max_lengthstridepad_to_multiple_ofreturn_token_type_idsreturn_attention_maskreturn_overflowing_tokensreturn_special_tokens_maskreturn_offsets_mappingreturn_lengthverbosereturn_tensorsreturnc                      | j                   d||||||||	|
||||||d|}| j                  ||      }|j                  |       |S )a  
        This method uses [`ViltImageProcessor.__call__`] method to prepare image(s) for the model, and
        [`BertTokenizerFast.__call__`] to prepare text for the model.

        Please refer to the docstring of the above two methods for more information.
        )r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   )r2    )r   r   update)r   imagesr$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r    encodingencoding_image_processors                       r"   __call__zViltProcessor.__call__@   sz    4 "4>> 
1!!1"7"7&?'A#9')
  !
& $(#7#7~#7#^ 01r#   c                 :     | j                   j                  |i |S )z
        This method forwards all its arguments to BertTokenizerFast's [`~PreTrainedTokenizer.batch_decode`]. Please
        refer to the docstring of this method for more information.
        )r   batch_decoder   argsr    s      r"   r<   zViltProcessor.batch_decoder   s     
 +t~~**D;F;;r#   c                 :     | j                   j                  |i |S )z
        This method forwards all its arguments to BertTokenizerFast's [`~PreTrainedTokenizer.decode`]. Please refer to
        the docstring of this method for more information.
        )r   decoder=   s      r"   r@   zViltProcessor.decodey   s     
 %t~~$$d5f55r#   c                     | j                   j                  }| j                  j                  }t        t        j                  ||z               S )N)r   model_input_namesr   listdictfromkeys)r   tokenizer_input_namesimage_processor_input_namess      r"   rB   zViltProcessor.model_input_names   s?     $ @ @&*&:&:&L&L#DMM"7:U"UVWWr#   c                 N    t        j                  dt               | j                  S )Nzg`feature_extractor_class` is deprecated and will be removed in v5. Use `image_processor_class` instead.)r   r   r   image_processor_classr   s    r"   feature_extractor_classz%ViltProcessor.feature_extractor_class   s"    u	
 )))r#   c                 N    t        j                  dt               | j                  S )Nz[`feature_extractor` is deprecated and will be removed in v5. Use `image_processor` instead.)r   r   r   r   rJ   s    r"   r   zViltProcessor.feature_extractor   s"    i	
 ###r#   )NN)NTFNNr   NNNFFFFTN)__name__
__module____qualname____doc__
attributesrI   tokenizer_classr   r   r   r
   r   boolstrr	   r   r   intr   r   r:   r<   r@   propertyrB   rK   r   __classcell__)r!   s   @r"   r   r      s    $[1J0<O6, _c#'5:;?$(,00404*/+0',#;?#0 I0$y/4HYCZZ[0 !	0
 tS/120 $%7780 SM0 0 %SM0  (~0  (~0 $(0 %)0 !%0 0  !0" !sJ!78#0& 
'0d<6 X X
 * * $ $r#   r   )rP   r   typingr   r   r   processing_utilsr   tokenization_utils_baser   r	   r
   r   r   utilsr   r   r5   r#   r"   <module>r\      s/     ( ( . w w y$N y$r#   