o
eiyR @ s8 d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
ddl m Z ddl
mZ dd lmZ dd
lmZ dd lmZmZmZmZmZ dd lmZ d
dlm Z d
dl m Z m Z m Z m!Z! d
dl"m#Z# d
dl$m%Z% d
dl&m'Z' d
dl(m)Z) e*e+Z,eg dZ-e e e-Z.de/fddZ0G dd dZ1ddgZ2dS )zAutoProcessor class. N) OrderedDict )PreTrainedConfig) get_class_from_dynamic_moduleresolve_trust_remote_code)FeatureExtractionMixin)ImageProcessingMixin)ProcessorMixin)TOKENIZER_CONFIG_FILE)FEATURE_EXTRACTOR_NAMEPROCESSOR_NAMEVIDEO_PROCESSOR_NAME cached_filelogging)BaseVideoProcessor )_LazyAutoMapping)CONFIG_MAPPING_NAMES
AutoConfigmodel_type_to_module_name!replace_list_option_in_docstrings)AutoFeatureExtractor)AutoImageProcessor)
AutoTokenizer)AutoVideoProcessor)})aimv2
CLIPProcessor)alignAlignProcessor)altclipAltCLIPProcessor)aria
AriaProcessor)audioflamingo3AudioFlamingo3Processor)
aya_visionAyaVisionProcessor)bark
BarkProcessor)blip
BlipProcessor)zblip-2Blip2Processor) bridgetowerBridgeTowerProcessor) chameleonChameleonProcessor) chinese_clipChineseCLIPProcessor)clap
ClapProcessor)clipr )clipsegCLIPSegProcessor)clvp
ClvpProcessor)cohere2_visionCohere2VisionProcessor)colpaliColPaliProcessor)colqwen2ColQwen2Processor) deepseek_vlDeepseekVLProcessor)deepseek_vl_hybridDeepseekVLHybridProcessor)dia DiaProcessor)edgetam
Sam2Processor)emu3
Emu3Processor)ernie4_5_vl_moeErnie4_5_VL_MoeProcessor)evollaEvollaProcessor)flavaFlavaProcessor) florence2Florence2Processor)fuyu
FuyuProcessor)gemma3Gemma3Processor)gemma3nGemma3nProcessor)git GitProcessor)glm46vGlm46VProcessor)glm4vGlm4vProcessor) glm4v_moer\ ) glm_imager\ )glmasrGlmAsrProcessor)got_ocr2GotOcr2Processor)granite_speechGraniteSpeechProcessor)zgrounding-dinoGroundingDinoProcessor)groupvitr )hubertWav2Vec2Processor)ideficsIdeficsProcessor)idefics2Idefics2Processor)idefics3Idefics3Processor) instructblipInstructBlipProcessor)instructblipvideoInstructBlipVideoProcessor)internvlInternVLProcessor)janusJanusProcessor)zkosmos-2Kosmos2Processor)z
kosmos-2.5Kosmos2_5Processor)kyutai_speech_to_textKyutaiSpeechToTextProcessor)lasr_ctc
LasrProcessor) lasr_encoderr| )
layoutlmv2LayoutLMv2Processor)
layoutlmv3LayoutLMv3Processor) layoutxlmLayoutXLMProcessor)lfm2_vlLfm2VlProcessor) lighton_ocrLightOnOcrProcessor)llama4Llama4Processor)llavaLlavaProcessor)
llava_nextLlavaNextProcessor)llava_next_videoLlavaNextVideoProcessor)llava_onevisionLlavaOnevisionProcessor)markuplmMarkupLMProcessor)
metaclip_2r )zmgp-strMgpstrProcessor)mistral3PixtralProcessor)mllamaMllamaProcessor)zmm-grounding-dinore ) moonshinerh )moonshine_streamingMoonshineStreamingProcessor)z omdet-turboOmDetTurboProcessor) oneformerOneFormerProcessor)ovis2Ovis2Processor)owlv2Owlv2Processor)owlvitOwlViTProcessor) paddleocr_vlPaddleOCRVLProcessor) paligemmaPaliGemmaProcessor)
perception_lmPerceptionLMProcessor)phi4_multimodalPhi4MultimodalProcessor)
pix2structPix2StructProcessor)pixtralr ) pop2pianoPop2PianoProcessor) qwen2_5_omniQwen2_5OmniProcessor)
qwen2_5_vlQwen2_5_VLProcessor) qwen2_audioQwen2AudioProcessor)qwen2_vlQwen2VLProcessor)qwen3_5Qwen3VLProcessor) qwen3_5_moer )qwen3_omni_moeQwen3OmniMoeProcessor)qwen3_vlr ) qwen3_vl_moer )sam SamProcessor)sam2rF )sam3
Sam3Processor)sam_hqSamHQProcessor) seamless_m4tSeamlessM4TProcessor)sewrh )zsew-drh ) shieldgemma2ShieldGemma2Processor)siglipSiglipProcessor)siglip2Siglip2Processor)smolvlmSmolVLMProcessor)speech_to_textSpeech2TextProcessor)speecht5SpeechT5Processor)t5gemma2rT )t5gemma2_encoderrT )trocrTrOCRProcessor)tvp TvpProcessor)udop
UdopProcessor) unispeechrh )z
unispeech-satrh ) video_llavaVideoLlavaProcessor)vilt
ViltProcessor)vipllavar )zvision-text-dual-encoder VisionTextDualEncoderProcessor)voxtralVoxtralProcessor)voxtral_realtimeVoxtralRealtimeProcessor)wav2vec2rh )z
wav2vec2-bertrh )zwav2vec2-conformerrh )wavlmrh )whisperWhisperProcessor)xclipXCLIPProcessor
class_namec C s t D ]'\}}| |v r+t|}td| d}zt|| W S ty* Y qw qtj D ]}t|dd | kr?| S q1td}t
|| rOt|| S d S )N.ztransformers.models__name__ transformers) PROCESSOR_MAPPING_NAMESitemsr importlib
import_modulegetattrAttributeErrorPROCESSOR_MAPPING_extra_contentvalueshasattr)r module_name
processorsmodule processor main_module r f/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/transformers/models/auto/processing_auto.pyprocessor_class_from_name s$
r c @ s: e Zd ZdZdd Zeeedd Ze d
ddZ
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 C s t d)Nz}AutoProcessor is designed to be instantiated using the `AutoProcessor.from_pretrained(pretrained_model_name_or_path)` method.)OSError)selfr r r __init__ s zAutoProcessor.__init__c s dd} dd}d d< d}d} fddttjD }|dddd t|tfi |}|durUtj|fi \} }
| d
}d | d i v rU| d d }|du rt|t
fi |} | durt j |fi \} }
| d
d}d | d i v r| d d }| du rt|t
fi |} | durtj|fi \} }
| d
d}d | d i v r| d d }| du rt|t
fi |} | dur|du rtj|fi \} }
| d
d}d | d i v r| d d }|du r-t|tfi |} | dur-t| d
d
}
t|
} W d n 1 sw Y | d
d}d | d i v r-| d d }|du rgz*t|tsDtj|fd|i }t|d
d}t|d r[d |j v r[|j d }W n
t yf Y nw |durpt |}|du}|dup~t |t v }|rd|v r|!dd }nd}t"|||||}|r|rt#||fi } dd}
|$ |j|fd|i S |dur|j|fd|i S t |t v rt t | j|fi S t%t&t't(fD ]}z|j|fd|i W S t)y Y qw t d| d)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.
Passing `token=True` is required when you want to use a private model.
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/")
```configNtrust_remote_codeT
_from_autoc s i | ] }| v r| | qS r r ).0keykwargsr r
# s z1AutoProcessor.from_pretrained..F) _raise_exceptions_for_gated_repo%_raise_exceptions_for_missing_entries'_raise_exceptions_for_connection_errorsprocessor_classr auto_mapzutf-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.)*popinspect signaturer
parametersupdater r get_processor_dictgetr r get_image_processor_dictr
r get_video_processor_dictr get_feature_extractor_dictr
openjsonload
isinstancer r from_pretrainedr r r
ValueErrorr typer splitr r register_for_auto_classr r r r Exception)cls pretrained_model_name_or_pathr r r r processor_auto_mapcached_file_kwargsprocessor_config_file config_dict_preprocessor_config_filetokenizer_config_filereaderhas_remote_codehas_local_code
upstream_repoklassr r r r( s B
z AutoProcessor.from_pretrainedFc C s t j| ||d dS )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< r r r r= s
zAutoProcessor.registerN)F) r
__module__ __qualname____doc__r
classmethodr r r( staticmethodr= r r r r r s Pr r )3rA 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__r r r r s:
m