
    쬜iyR                     f   d 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' ddl(m)Z)  ejT                  e+      Z, eg d      Z- eee-      Z.de/fdZ0 G d d      Z1ddgZ2y)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)AutoVideoProcessor)})aimv2CLIPProcessor)alignAlignProcessor)altclipAltCLIPProcessor)ariaAriaProcessor)audioflamingo3AudioFlamingo3Processor)
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)ernie4_5_vl_moeErnie4_5_VL_MoeProcessor)evollaEvollaProcessor)flavaFlavaProcessor)	florence2Florence2Processor)fuyuFuyuProcessor)gemma3Gemma3Processor)gemma3nGemma3nProcessor)gitGitProcessor)glm46vGlm46VProcessor)glm4vGlm4vProcessor)	glm4v_moer]   )	glm_imager]   )glmasrGlmAsrProcessor)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)lasr_ctcLasrProcessor)lasr_encoderr}   )
layoutlmv2LayoutLMv2Processor)
layoutlmv3LayoutLMv3Processor)	layoutxlmLayoutXLMProcessor)lfm2_vlLfm2VlProcessor)lighton_ocrLightOnOcrProcessor)llama4Llama4Processor)llavaLlavaProcessor)
llava_nextLlavaNextProcessor)llava_next_videoLlavaNextVideoProcessor)llava_onevisionLlavaOnevisionProcessor)markuplmMarkupLMProcessor)
metaclip_2r   )zmgp-strMgpstrProcessor)mistral3PixtralProcessor)mllamaMllamaProcessor)zmm-grounding-dinorf   )	moonshineri   )moonshine_streamingMoonshineStreamingProcessor)zomdet-turboOmDetTurboProcessor)	oneformerOneFormerProcessor)ovis2Ovis2Processor)owlv2Owlv2Processor)owlvitOwlViTProcessor)paddleocr_vlPaddleOCRVLProcessor)	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_5Qwen3VLProcessor)qwen3_5_moer   )qwen3_omni_moeQwen3OmniMoeProcessor)qwen3_vlr   )qwen3_vl_moer   )samSamProcessor)sam2rG   )sam3Sam3Processor)sam_hqSamHQProcessor)seamless_m4tSeamlessM4TProcessor)sewri   )zsew-dri   )shieldgemma2ShieldGemma2Processor)siglipSiglipProcessor)siglip2Siglip2Processor)smolvlmSmolVLMProcessor)speech_to_textSpeech2TextProcessor)speecht5SpeechT5Processor)t5gemma2rU   )t5gemma2_encoderrU   )trocrTrOCRProcessor)tvpTvpProcessor)udopUdopProcessor)	unispeechri   )zunispeech-satri   )video_llavaVideoLlavaProcessor)viltViltProcessor)vipllavar   )zvision-text-dual-encoderVisionTextDualEncoderProcessor)voxtralVoxtralProcessor)voxtral_realtimeVoxtralRealtimeProcessor)wav2vec2ri   )zwav2vec2-bertri   )zwav2vec2-conformerri   )wavlmri   )whisperWhisperProcessor)xclipXCLIPProcessor
class_namec                    t         j                         D ]<  \  }}| |v st        |      }t        j                  d| d      }	 t        ||       c S  t        j                  j                         D ]  }t        |dd       | k(  s|c S  t        j                  d      }t        ||       rt        ||       S y # t        $ r Y w xY w)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         d/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/transformers/models/auto/processing_auto.pyprocessor_class_from_namer     s    #:#@#@#B Z#3K@K,,q->@UVFvz22 '55<<> 	9j$/:= )).9K{J'{J// " s   B99	CCc                   N    e Zd ZdZd Ze ee      d               Ze	dd       Z
y)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                 &   |j                  dd      }|j                  dd      }d|d<   d}d}t        j                  t              j                  D ci c]  }||v s|||    }}|j                  dddd       t        |t        fi |}	|	Ft        j                  |fi |\  }
}|
j                  d      }d	|
j                  d
i       v r|
d
   d	   }|t        |t        fi |}|Gt        j                  |fi |\  }
}|
j                  dd      }d	|
j                  d
i       v r|
d
   d	   }|Zt        |t        fi |}|Gt        j                  |fi |\  }
}|
j                  dd      }d	|
j                  d
i       v r|
d
   d	   }|\t        |t        fi |}|I|Gt!        j"                  |fi |\  }
}|
j                  dd      }d	|
j                  d
i       v r|
d
   d	   }|lt        |t$        fi |}|Yt'        |d      5 }t)        j*                  |      }
ddd       
j                  dd      }d	|
j                  d
i       v r|
d
   d	   }|_	 t-        |t.              st1        j2                  |fd|i|}t5        |dd      }t7        |d
      rd	|j8                  v r|j8                  d	   }|t=        |      }|du}|duxs t?        |      t@        v }|r*d|v r|jC                  d      d   }nd}tE        |||||      }|rF|rDtG        ||fi |}|j                  dd      }|jI                           |j2                  |fd|i|S | |j2                  |fd|i|S t?        |      t@        v r"tA        t?        |         j2                  |fi |S tJ        tL        tN        tP        fD ]  }	  |j2                  |fd|i|c S  t;        d| d      c c}w # 1 sw Y   xY w# t:        $ r Y Ew xY w# tR        $ r Y Xw xY w)a3  
        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.
            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/")
        ```configNtrust_remote_codeT
_from_autoF) _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, a video processor or a feature extractor for this model. Make sure the repository contains the files of at least one of those processing classes.)*popinspect	signaturer   
parametersupdater   r
   get_processor_dictgetr   r	   get_image_processor_dictr   r   get_video_processor_dictr   get_feature_extractor_dictr   openjsonload
isinstancer   r   from_pretrainedr   r   r  
ValueErrorr  typer   splitr   r   register_for_auto_classr   r   r   r   	Exception)clspretrained_model_name_or_pathkwargsr  r  r  processor_auto_mapkeycached_file_kwargsprocessor_config_fileconfig_dict_preprocessor_config_filetokenizer_config_filereaderhas_remote_codehas_local_codeupstream_repoklasss                      r  r(  zAutoProcessor.from_pretrained   sE   D Hd+"JJ':DA#|! ;B:K:KK:X:c:cu3gjntgtc6#;.uu!!499>;@	
 !,,I> p]o p ,+>>?\g`fgNK)oo.?@O+//*b"AA%0%<_%M""'2-/E(I[($ (3!5!N!NOl!wpv!wQ"-//2CT"J"kooj"&EE)4Z)@)Q& (/+613G,K],( ,7%7%P%P5&9?&NK '2oo6G&NO&+//*b*II-8-D_-U*'/+613I,M_,( ,7O<S%;%V%V5&9?&NK '2oo6G&NO&+//*b*II-8-D_-U*"$/-/D%HZ%! %0/'B 4f"&))F"3K4 #.//2CT"J"kooj"&EE)4Z)@)Q&"!&*:;'775IZ^dF #*&2CT"J6:.?foo3U)/)I& &7HO,D8(4YVHY8Y)) 2 8 8 >q A $ 9!#@.Racp! 0;"$AEKO 

?D1A3352?22-ARV\  (2?22-ARV\  &\..$T&\2BBC`kdjkk $%79KMab 	E,u,,1EVZ` 	 /0M/N Og g
 	
A vx4 4,   V  s=   	O"O"O'AO4 ;P'O14	P P	PPc                 4    t         j                  | ||       y)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  r@  zAutoProcessor.register  s     	""<8"Tr  N)F)r   
__module____qualname____doc__r  classmethodr   r   r(  staticmethodr@   r  r  r	  r	     sH    
 &'>?M
 @ M
^ 	U 	Ur  r	  r   )3rD  r   r  r%  collectionsr   configuration_utilsr   dynamic_module_utilsr   r   feature_extraction_utilsr   image_processing_utilsr	   processing_utilsr
   tokenization_pythonr   utilsr   r   r   r   r   video_processing_utilsr   auto_factoryr   configuration_autor   r   r   r   feature_extraction_autor   image_processing_autor   tokenization_autor   video_processing_autor   
get_loggerr   loggerr   r   strr  r	  __all__rG  r  r  <module>r[     s        # 4 \ > : . 8 g g 8 *  : 5 , 5 
		H	%%~@ D %%9;RS # 0iU iUX 
0r  