
    Z jG                         S r SSKJr  SSKJrJrJr  SSKJrJ	r	J
r
  SSKJrJr   " S S\SS	9r\R                  " \5      r\ " S
 S\5      5       rS/rg)z!
Processor class for Pix2Struct.
   )BatchFeature)ProcessingKwargsProcessorMixinUnpack)BatchEncodingPreTokenizedInput	TextInput)auto_docstringloggingc            
       6    \ rS rSrSSSSSSSSSS.	SS0S.rS	rg
)Pix2StructProcessorKwargs   TF    )	add_special_tokenspaddingstridereturn_overflowing_tokensreturn_special_tokens_maskreturn_offsets_mappingreturn_token_type_idsreturn_lengthverbosemax_patchesi   )text_kwargsimages_kwargs N)__name__
__module____qualname____firstlineno__	_defaults__static_attributes__r       څ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/pix2struct/processing_pix2struct.pyr   r      s7     #').*/&+%*"

 4
Ir#   r   F)totalc            	          ^  \ rS rSrU 4S jr\  S	S\\-  \\   -  \\   -  S\	\
   S\\-  4S jj5       r\S 5       rSrU =r$ )
Pix2StructProcessor.   c                 2   > SUl         [        TU ]	  X5        g )NF)r   super__init__)selfimage_processor	tokenizer	__class__s      r$   r+   Pix2StructProcessor.__init__0   s    */	'4r#   textkwargsreturnc                    Uc  Uc  [        S5      eU R                  " [        4SU R                  R                  0UD6nUS   R                  SS 5      nUcA  U R                  R                  (       d&  Ub  UOSUS   S'   U R                  " SSU0US   D6nU$ U R                  R                  (       d  U R                  " U40 US   D6nO+US   R                  SU5        U R                  " U40 US   D6nUbt  U R                  R                  (       dY  Ub  UOS	US   S'   U R                  " SSU0US   D6nS
U;   a  UR                  S
5      US'   SU;   a  UR                  S5      US'   OS nUb  UR                  U5        U$ )Nz*You have to specify either images or text.tokenizer_init_kwargsr   r   Tr1   r   header_textFattention_maskdecoder_attention_mask	input_idsdecoder_input_idsr   )

ValueError_merge_kwargsr   r.   init_kwargspopr-   is_vqa
setdefaultupdate)r,   imagesr1   r2   output_kwargsr   text_encodingencoding_image_processors           r$   __call__Pix2StructProcessor.__call__4   s    >dlIJJ**%
"&.."<"<
 

 +=9==>RTXY>$"6"6"="=&8&D"$ -()=> !NNUUm8TUM  ##**'+';';F'emTcFd'e$ /*55mTJ'+';';F'emTcFd'e$D$8$8$?$?&8&D"% -()=> !NNUUm8TUM=0:G:K:KL\:]67m+5B5F5F{5S12 M$$++M:''r#   c                 >    U R                   R                  nSS/nX-   $ )Nr8   r:   )r-   model_input_names)r,   image_processor_input_namesdecoder_idss      r$   rI   %Pix2StructProcessor.model_input_namesf   s(    &*&:&:&L&L#/1DE*88r#   r   )NN)r   r   r   r    r+   r
   r	   r   listr   r   r   r   rF   propertyrI   r"   __classcell__)r/   s   @r$   r'   r'   .   s|    5  Z^/( ++d9o=EV@WW/( 23	/(
 
	%/( /(b 9 9r#   r'   N)__doc__feature_extraction_utilsr   processing_utilsr   r   r   tokenization_utils_baser   r   r	   utilsr
   r   r   
get_loggerr   loggerr'   __all__r   r#   r$   <module>rX      se    5 H H R R , 0 & 
		H	% ;9. ;9 ;9| !
!r#   