o
iuQ @ s4 d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
ddlmZ dd lmZ dd
lmZ dd lmZmZmZmZmZ dd lmZ d
dl m Z d
dl m Z m Z m!Z!m"Z" d
dl#m$Z$ d
dl%m&Z& d
dl'm(Z( e)e*Z+eg dZ,e e e,Z-de.fddZ/G dd dZ0ddgZ1dS )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)n)aimv2
CLIPProcessor)alignAlignProcessor)altclipAltCLIPProcessor)aria
AriaProcessor)
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)evollaEvollaProcessor)flavaFlavaProcessor) florence2Florence2Processor)fuyu
FuyuProcessor)gemma3Gemma3Processor)gemma3nGemma3nProcessor)git GitProcessor)glm4vGlm4vProcessor) glm4v_moerU )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)
layoutlmv2LayoutLMv2Processor)
layoutlmv3LayoutLMv3Processor)lfm2_vlLfm2VlProcessor)llama4Llama4Processor)llavaLlavaProcessor)
llava_nextLlavaNextProcessor)llava_next_videoLlavaNextVideoProcessor)llava_onevisionLlavaOnevisionProcessor)markuplmMarkupLMProcessor)mctctMCTCTProcessor)
metaclip_2r )zmgp-strMgpstrProcessor)mistral3PixtralProcessor)mllamaMllamaProcessor)zmm-grounding-dinor[ ) moonshiner^ ) oneformerOneFormerProcessor)ovis2Ovis2Processor)owlv2Owlv2Processor)owlvitOwlViTProcessor) paligemmaPaliGemmaProcessor)
perception_lmPerceptionLMProcessor)phi4_multimodalPhi4MultimodalProcessor)
pix2structPix2StructProcessor)pixtralr ) pop2pianoPop2PianoProcessor) qwen2_5_omniQwen2_5OmniProcessor)
qwen2_5_vlQwen2_5_VLProcessor) qwen2_audioQwen2AudioProcessor)qwen2_vlQwen2VLProcessor)qwen3_omni_moeQwen3OmniMoeProcessor)qwen3_vlQwen3VLProcessor) qwen3_vl_moer )sam SamProcessor)sam2rC )sam_hqSamHQProcessor) seamless_m4tSeamlessM4TProcessor)sewr^ )zsew-dr^ ) shieldgemma2ShieldGemma2Processor)siglipSiglipProcessor)siglip2Siglip2Processor)smolvlmSmolVLMProcessor)speech_to_textSpeech2TextProcessor)speech_to_text_2Speech2Text2Processor)speecht5SpeechT5Processor)trocrTrOCRProcessor)tvlt
TvltProcessor)tvp TvpProcessor)udop
UdopProcessor) unispeechr^ )z
unispeech-satr^ ) video_llavaVideoLlavaProcessor)vilt
ViltProcessor)vipllavarz )zvision-text-dual-encoder VisionTextDualEncoderProcessor)voxtralVoxtralProcessor)wav2vec2r^ )z
wav2vec2-bertr^ )zwav2vec2-conformerr^ )wavlmr^ )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 e/home/ubuntu/veenaModal/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}|dur tdt ddurtd| d< dd} dd}d d < d}d} fd
d ttj D }|
d d d d
t|t fi |} | durut j
|fi \}
} |
dd}d|
di v ru|
d d }|du rt|tfi |} | durtj|fi \}
} |
dd}d|
di v r|
d d }| du rt|tfi |} | durtj|fi \}
} |
dd}d|
di v r|
d d }| du rt|tfi |} | dur|du rtj|fi \}
} |
dd}d|
di v r|
d d }|du rRt|tfi |}
|
durRt|
dd
}t|}
W d n 1 s8w Y |
dd}d|
di v rR|
d d }|du rt|tsht j |fd|i }t |dd}t |drd|j v r|j d }|durt!|}|du}|dupt"|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 z t(j |fd|i W S t)yA zt*j |fd|i W Y S t)y% Y nw zt+j |fd|i W Y S t)y> Y nw Y nw td| 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.
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/")
```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`.configtrust_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 or a feature extractor for this model. Make sure the repository contains the files of at least one of those processing classes.),popwarningswarn
FutureWarningget
ValueErrorinspect signaturer
parametersupdater r get_processor_dictr r get_image_processor_dictr
r get_video_processor_dictr get_feature_extractor_dictr
openjsonload
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_mapcached_file_kwargsprocessor_config_file config_dict_preprocessor_config_filetokenizer_config_filereaderhas_remote_codehas_local_code
upstream_repor r r r sN E
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 cr r )2r- 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 r s:
s