
     `iuQ                     r   d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl	m
Z
mZ ddlmZ ddlmZ dd	l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 ddlmZ ddlmZm Z m!Z!m"Z" ddl#m$Z$ ddl%m&Z& ddl'm(Z(  ej)        e*          Z+ eg d          Z, eee,          Z-de.fdZ/ G d d          Z0ddgZ1dS )zAutoProcessor class.    N)OrderedDict   )PretrainedConfig)get_class_from_dynamic_moduleresolve_trust_remote_code)FeatureExtractionMixin)ImageProcessingMixin)ProcessorMixin)TOKENIZER_CONFIG_FILE)FEATURE_EXTRACTOR_NAMEPROCESSOR_NAMEVIDEO_PROCESSOR_NAMEcached_filelogging)BaseVideoProcessor   )_LazyAutoMapping)CONFIG_MAPPING_NAMES
AutoConfigmodel_type_to_module_name!replace_list_option_in_docstrings)AutoFeatureExtractor)AutoImageProcessor)AutoTokenizer)n)aimv2CLIPProcessor)alignAlignProcessor)altclipAltCLIPProcessor)ariaAriaProcessor)
aya_visionAyaVisionProcessor)barkBarkProcessor)blipBlipProcessor)zblip-2Blip2Processor)bridgetowerBridgeTowerProcessor)	chameleonChameleonProcessor)chinese_clipChineseCLIPProcessor)clapClapProcessor)clipr   )clipsegCLIPSegProcessor)clvpClvpProcessor)cohere2_visionCohere2VisionProcessor)colpaliColPaliProcessor)colqwen2ColQwen2Processor)deepseek_vlDeepseekVLProcessor)deepseek_vl_hybridDeepseekVLHybridProcessor)diaDiaProcessor)edgetamSam2Processor)emu3Emu3Processor)evollaEvollaProcessor)flavaFlavaProcessor)	florence2Florence2Processor)fuyuFuyuProcessor)gemma3Gemma3Processor)gemma3nGemma3nProcessor)gitGitProcessor)glm4vGlm4vProcessor)	glm4v_moerV   )got_ocr2GotOcr2Processor)granite_speechGraniteSpeechProcessor)zgrounding-dinoGroundingDinoProcessor)groupvitr   )hubertWav2Vec2Processor)ideficsIdeficsProcessor)idefics2Idefics2Processor)idefics3Idefics3Processor)instructblipInstructBlipProcessor)instructblipvideoInstructBlipVideoProcessor)internvlInternVLProcessor)janusJanusProcessor)zkosmos-2Kosmos2Processor)z
kosmos-2.5Kosmos2_5Processor)kyutai_speech_to_textKyutaiSpeechToTextProcessor)
layoutlmv2LayoutLMv2Processor)
layoutlmv3LayoutLMv3Processor)lfm2_vlLfm2VlProcessor)llama4Llama4Processor)llavaLlavaProcessor)
llava_nextLlavaNextProcessor)llava_next_videoLlavaNextVideoProcessor)llava_onevisionLlavaOnevisionProcessor)markuplmMarkupLMProcessor)mctctMCTCTProcessor)
metaclip_2r   )zmgp-strMgpstrProcessor)mistral3PixtralProcessor)mllamaMllamaProcessor)zmm-grounding-dinor\   )	moonshiner_   )	oneformerOneFormerProcessor)ovis2Ovis2Processor)owlv2Owlv2Processor)owlvitOwlViTProcessor)	paligemmaPaliGemmaProcessor)perception_lmPerceptionLMProcessor)phi4_multimodalPhi4MultimodalProcessor)
pix2structPix2StructProcessor)pixtralr   )	pop2pianoPop2PianoProcessor)qwen2_5_omniQwen2_5OmniProcessor)
qwen2_5_vlQwen2_5_VLProcessor)qwen2_audioQwen2AudioProcessor)qwen2_vlQwen2VLProcessor)qwen3_omni_moeQwen3OmniMoeProcessor)qwen3_vlQwen3VLProcessor)qwen3_vl_moer   )samSamProcessor)sam2rD   )sam_hqSamHQProcessor)seamless_m4tSeamlessM4TProcessor)sewr_   )zsew-dr_   )shieldgemma2ShieldGemma2Processor)siglipSiglipProcessor)siglip2Siglip2Processor)smolvlmSmolVLMProcessor)speech_to_textSpeech2TextProcessor)speech_to_text_2Speech2Text2Processor)speecht5SpeechT5Processor)trocrTrOCRProcessor)tvltTvltProcessor)tvpTvpProcessor)udopUdopProcessor)	unispeechr_   )zunispeech-satr_   )video_llavaVideoLlavaProcessor)viltViltProcessor)vipllavar{   )zvision-text-dual-encoderVisionTextDualEncoderProcessor)voxtralVoxtralProcessor)wav2vec2r_   )zwav2vec2-bertr_   )zwav2vec2-conformerr_   )wavlmr_   )whisperWhisperProcessor)xclipXCLIPProcessor
class_namec                    t                                           D ]S\  }}| |v rJt          |          }t          j        d| d          }	 t          ||           c S # t          $ r Y Ow xY wTt          j        	                                D ]}t          |dd           | k    r|c S t          j        d          }t          ||           rt          ||           S d S )N.ztransformers.models__name__transformers)PROCESSOR_MAPPING_NAMESitemsr   	importlibimport_modulegetattrAttributeErrorPROCESSOR_MAPPING_extra_contentvalueshasattr)r   module_name
processorsmodule	processormain_modules         |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/transformers/models/auto/processing_auto.pyprocessor_class_from_namer      s   #:#@#@#B#B  Z##3K@@K,->->->@UVVFvz22222!    $ '5<<>>  	9j$//:== >
 ).99K{J'' 0{J///4s   A
A*)A*c                   h    e Zd ZdZd Ze ee          d                         Ze	dd            Z
dS )AutoProcessora  
    This is a generic processor class that will be instantiated as one of the processor classes of the library when
    created with the [`AutoProcessor.from_pretrained`] class method.

    This class cannot be instantiated directly using `__init__()` (throws an error).
    c                      t          d          )Nz}AutoProcessor is designed to be instantiated using the `AutoProcessor.from_pretrained(pretrained_model_name_or_path)` method.)OSError)selfs    r   __init__zAutoProcessor.__init__   s    _
 
 	
    c                 	                        dd          }|Ct          j        dt                                         d          t          d          |d<                        dd          }                     dd          }dd	<   d}d}fd
t          j        t                    j	        D             }|
                    dddd           t          |t          fi |}	|	Qt          j        |fi \  }
}|
                    dd          }d|
                    di           v r|
d         d         }|8t          |t          fi |}|Qt          j        |fi \  }
}|
                    dd          }d|
                    di           v r|
d         d         }|ft          |t"          fi |}|Qt%          j        |fi \  }
}|
                    dd          }d|
                    di           v r|
d         d         }|ht          |t          fi |}|S|Qt)          j        |fi \  }
}|
                    dd          }d|
                    di           v r|
d         d         }|t          |t,          fi |}|yt/          |d          5 }t1          j        |          }
ddd           n# 1 swxY w Y   |
                    dd          }d|
                    di           v r|
d         d         }|`t5          |t6                    st9          j        |fd|i}t=          |dd          }t?          |d          rd|j         v r|j         d         }|tC          |          }|du}|duptE          |          tF          v }|r5d|v r|$                    d          d         }nd}tK          |||||          }|rJ|rHtM          ||fi }                     dd          }|'                                  |j        |fd|iS | |j        |fd|iS tE          |          tF          v r%tG          tE          |                   j        |fi S 	 tQ          j        |fd|iS # tR          $ rQ 	 tU          j        |fd|icY S # tR          $ r Y nw xY w	 tW          j        |fd|icY S # tR          $ r Y nw xY wY nw xY wt          d| d          )a  
        Instantiate one of the processor classes of the library from a pretrained model vocabulary.

        The processor class to instantiate is selected based on the `model_type` property of the config object (either
        passed as an argument or loaded from `pretrained_model_name_or_path` if possible):

        List options

        Params:
            pretrained_model_name_or_path (`str` or `os.PathLike`):
                This can be either:

                - a string, the *model id* of a pretrained feature_extractor hosted inside a model repo on
                  huggingface.co.
                - a path to a *directory* containing a processor files saved using the `save_pretrained()` method,
                  e.g., `./my_model_directory/`.
            cache_dir (`str` or `os.PathLike`, *optional*):
                Path to a directory in which a downloaded pretrained model feature extractor should be cached if the
                standard cache should not be used.
            force_download (`bool`, *optional*, defaults to `False`):
                Whether or not to force to (re-)download the feature extractor files and override the cached versions
                if they exist.
            resume_download:
                Deprecated and ignored. All downloads are now resumed by default when possible.
                Will be removed in v5 of Transformers.
            proxies (`dict[str, str]`, *optional*):
                A dictionary of proxy servers to use by protocol or endpoint, e.g., `{'http': 'foo.bar:3128',
                'http://hostname': 'foo.bar:4012'}.` The proxies are used on each request.
            token (`str` or *bool*, *optional*):
                The token to use as HTTP bearer authorization for remote files. If `True`, will use the token generated
                when running `hf auth login` (stored in `~/.huggingface`).
            revision (`str`, *optional*, defaults to `"main"`):
                The specific model version to use. It can be a branch name, a tag name, or a commit id, since we use a
                git-based system for storing models and other artifacts on huggingface.co, so `revision` can be any
                identifier allowed by git.
            return_unused_kwargs (`bool`, *optional*, defaults to `False`):
                If `False`, then this function returns just the final feature extractor object. If `True`, then this
                functions returns a `Tuple(feature_extractor, unused_kwargs)` where *unused_kwargs* is a dictionary
                consisting of the key/value pairs whose keys are not feature extractor attributes: i.e., the part of
                `kwargs` which has not been used to update `feature_extractor` and is otherwise ignored.
            trust_remote_code (`bool`, *optional*, defaults to `False`):
                Whether or not to allow for custom models defined on the Hub in their own modeling files. This option
                should only be set to `True` for repositories you trust and in which you have read the code, as it will
                execute code present on the Hub on your local machine.
            kwargs (`dict[str, Any]`, *optional*):
                The values in kwargs of any keys which are feature extractor attributes will be used to override the
                loaded values. Behavior concerning key/value pairs whose keys are *not* feature extractor attributes is
                controlled by the `return_unused_kwargs` keyword parameter.

        <Tip>

        Passing `token=True` is required when you want to use a private model.

        </Tip>

        Examples:

        ```python
        >>> from transformers import AutoProcessor

        >>> # Download processor from huggingface.co and cache.
        >>> processor = AutoProcessor.from_pretrained("facebook/wav2vec2-base-960h")

        >>> # If processor files are in a directory (e.g. processor was saved using *save_pretrained('./test/saved_model/')*)
        >>> # processor = AutoProcessor.from_pretrained("./test/saved_model/")
        ```use_auth_tokenNzrThe `use_auth_token` argument is deprecated and will be removed in v5 of Transformers. Please use `token` instead.tokenzV`token` and `use_auth_token` are both specified. Please set only the argument `token`.configtrust_remote_codeT
_from_autoc                 *    i | ]}|v ||         S  r   ).0keykwargss     r   
<dictcomp>z1AutoProcessor.from_pretrained.<locals>.<dictcomp>$  s+    uuu3gjntgtgtc6#;gtgtgtr   F) _raise_exceptions_for_gated_repo%_raise_exceptions_for_missing_entries'_raise_exceptions_for_connection_errorsprocessor_classr   auto_mapzutf-8)encodingz--r   code_revisionz!Unrecognized processing class in z. Can't instantiate a processor, a tokenizer, an image processor or a feature extractor for this model. Make sure the repository contains the files of at least one of those processing classes.),popwarningswarnFutureWarningget
ValueErrorinspect	signaturer   
parametersupdater   r
   get_processor_dictr   r	   get_image_processor_dictr   r   get_video_processor_dictr   get_feature_extractor_dictr   openjsonload
isinstancer   r   from_pretrainedr   r   r  r   typer   splitr   r   register_for_auto_classr   	Exceptionr   r   )clspretrained_model_name_or_pathr  r   r   r   r  processor_auto_mapcached_file_kwargsprocessor_config_fileconfig_dict_preprocessor_config_filetokenizer_config_filereaderhas_remote_codehas_local_codeupstream_repos     `               r   r  zAutoProcessor.from_pretrained   so   J  $4d;;%M E   zz'"". l   -F7OHd++"JJ':DAA#|! vuuu':KK:X:X:cuuu!!499>;@ 	
 	
 	
 !,,I> p p]o p p ,+>?\gg`fggNK)oo.?FFO+//*b"A"AAA%0%<_%M""'2-/E( (I[( ($ (3!5!NOl!w!wpv!w!wQ"-//2CT"J"J"kooj"&E&EEE)4Z)@)Q& (/+613G, ,K], ,( ,7%7%P5& &9?& &NK '2oo6G&N&NO&+//*b*I*III-8-D_-U* (/+613I, ,M_, ,( ,7O<S%;%V5& &9?& &NK '2oo6G&N&NO&+//*b*I*III-8-D_-U*"$/-/D% %HZ% %! %0/'BBB 4f"&)F"3"3K4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 #.//2CT"J"J"kooj"&E&EEE)4Z)@)Q&"f&677 #31 EVZ` 
 &f.?FFOvz** F&//Q/Q%+__%E"&7HHO,D8(4YVHY8Y 	))) 2 8 8 > >q A $ 9!#@.Racp! !  	l0 	l;"$A EK O 

?D11A335552?2- ARV\   (2?2- ARV\   &\\...$T&\\2BC`kkdjkkk	 0- ARV\    	 	 	)91 EVZ`        +;1 EVZ`        	 E0M E E E
 
 	
sl   J??KK(Q< <
SRS
R*'S)R**S.SS
SSSSSFc                 @    t                               | ||           dS )a  
        Register a new processor for this class.

        Args:
            config_class ([`PretrainedConfig`]):
                The configuration corresponding to the model to register.
            processor_class ([`ProcessorMixin`]): The processor to register.
        )exist_okN)r   register)config_classr  r/  s      r   r0  zAutoProcessor.register  s%     	""<8"TTTTTr   N)F)r   
__module____qualname____doc__r   classmethodr   r   r  staticmethodr0  r   r   r   r   r      s         
 
 
 &&'>??`
 `
 @? [`
D 	U 	U 	U \	U 	U 	Ur   r   r   )2r4  r   r  r  r  collectionsr   configuration_utilsr   dynamic_module_utilsr   r   feature_extraction_utilsr   image_processing_utilsr	   processing_utilsr
   tokenization_utilsr   utilsr   r   r   r   r   video_processing_utilsr   auto_factoryr   configuration_autor   r   r   r   feature_extraction_autor   image_processing_autor   tokenization_autor   
get_loggerr   loggerr   r   strr   r   __all__r   r   r   <module>rI     s2            # # # # # # 4 3 3 3 3 3 \ \ \ \ \ \ \ \ > > > > > > : : : : : : . . . . . . 7 7 7 7 7 7 g g g g g g g g g g g g g g 8 8 8 8 8 8 * * * * * *            : 9 9 9 9 9 5 5 5 5 5 5 , , , , , , 
	H	%	%%+o o oq q f %$%9;RSS #    0|U |U |U |U |U |U |U |U~ 
0r   