^jO U d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
mZmZmZmZmZmZmZ dd lmZ dd
lmZ d d lmZ d d
lm Z d dl m Z m Z m Z m!Z! ejD e# Z$er e Z ee%e&e%dz e%dz f f e'd< n eg d Z( e jR e( e jT D ] \ Z+Z,e,Z- e sdZ-e-e e+< ee e Z.de%fdZ/ d de%ej` z de%ej` z dz de1de2e%e%f dz de1e%z dz de%dz de1fdZ3 ed G d d Z4d d gZ5y)!zAutoVideoProcessor class. N) OrderedDict)
TYPE_CHECKING )PreTrainedConfig) get_class_from_dynamic_moduleresolve_trust_remote_code) CONFIG_NAMEIMAGE_PROCESSOR_NAMEPROCESSOR_NAMEVIDEO_PROCESSOR_NAME cached_fileis_torchvision_availableloggingsafe_load_json_file)requires)BaseVideoProcessor )_LazyAutoMapping) VIDEO_PROCESSOR_MAPPING_NAMES)CONFIG_MAPPING_NAMES
AutoConfigmodel_type_to_module_name!replace_list_option_in_docstringsr ) ) cosmos3_omniQwen3VLVideoProcessor) exaone4_5Qwen2VLVideoProcessor) instructblip InstructBlipVideoVideoProcessor)pe_audio_videoPeVideoVideoProcessor) qwen2_5_omnir )
qwen2_5_vlr )qwen3_5r ) qwen3_5_moer )qwen3_omni_moer ) qwen3_vl_moer )
videoprism LlavaOnevisionVideoProcessor
class_namec t j D ]= \ }}| |k( s t | }t j d| d } t || c S t j j D ] }t |dd | k( s|c S t j d }t || r t || S y # t $ r Y w xY w)N.ztransformers.models__name__ transformers) r itemsr importlib
import_modulegetattrAttributeErrorVIDEO_PROCESSOR_MAPPING_extra_contentvalueshasattr)r* module_name extractormodule main_modules y/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/transformers/models/auto/video_processing_auto.py video_processor_class_from_namer= T s "?"E"E"G Y "3K@K ,,q
->@UVF
vz22
-;;BBD 9j$ /: = )).9K{J'{J// "
s B:: CC pretrained_model_name_or_path cache_dirforce_downloadproxiestokenrevisionlocal_files_onlyc < t | t ||||||dd
}t t fD cg c] } t | | ||||||ddd x}
|
} } | r| d nd} | s|st j d i S i }
|t
| }d|v r|d }
|
|
t
| }
|
S c c} w )a
Loads the video processor configuration from a pretrained model video processor configuration.
Args:
pretrained_model_name_or_path (`str` or `os.PathLike`):
This can be either:
- a string, the *model id* of a pretrained model configuration hosted inside a model repo on
huggingface.co.
- a path to a *directory* containing a configuration file saved using the
[`~BaseVideoProcessor.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 configuration 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 configuration 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.
local_files_only (`bool`, *optional*, defaults to `False`):
If `True`, will only try to load the video processor configuration from local files.
Passing `token=True` is required when you want to use a private model.
Returns:
`Dict`: The configuration of the video processor.
Examples:
```python
# Download configuration from huggingface.co and cache.
video_processor_config = get_video_processor_config("llava-hf/llava-onevision-qwen2-0.5b-ov-hf")
# This model does not have a video processor config so the result will be an empty dict.
video_processor_config = get_video_processor_config("FacebookAI/xlm-roberta-base")
# Save a pretrained video processor locally and you can reload its config
from transformers import AutoVideoProcessor
video_processor = AutoVideoProcessor.from_pretrained("llava-hf/llava-onevision-qwen2-0.5b-ov-hf")
video_processor.save_pretrained("video-processor-test")
video_processor = get_video_processor_config("video-processor-test")
```F) filenamer? r@ rA rB rC rD _raise_exceptions_for_gated_repo%_raise_exceptions_for_missing_entries)
rF r? r@ rA rB rC rD rG rH '_raise_exceptions_for_connection_errorsNr z8Could not locate the video processor configuration file.video_processor)r
r r r
loggerinfor )r> r? r@ rA rB rC rD kwargsresolved_processor_filerF
resolved_file resolved_video_processor_files resolved_video_processor_filevideo_processor_dictprocessor_dicts r< get_video_processor_configrT l s D *% %))..3 ./CD& (-!# - !!1166;8=
M &" &( Jh$B1$Emq! )1H NO
*,-DE .#12C#D $05I5Q 23P Q O&s "B)vision torchvision)backendsc P e Zd ZdZd Ze ee d Ze dd Z
y)AutoVideoProcessora%
This is a generic video processor class that will be instantiated as one of the video processor classes of the
library when created with the [`AutoVideoProcessor.from_pretrained`] class method.
This class cannot be instantiated directly using `__init__()` (throws an error).
c t d )NzAutoVideoProcessor is designed to be instantiated using the `AutoVideoProcessor.from_pretrained(pretrained_model_name_or_path)` method.)OSError)selfs r< __init__zAutoVideoProcessor.__init__ s
d
c |j dd }|j dd }d|d< t j |fi |\ }}|j dd }d} d|j di v r|d d } |c| a|j d d }
|
|
j d
d } t | | }d |j di v r|d d } | j d
d } |_ t
|t st j |fd|i|}t |dd }t |d r d|j v r|j d } | t | }| du}
|duxs t | t v }|xr0 |xs t t | j j# d
}|
r*d| v r| j% d d }nd}t' ||||
| }|
rK|rI|sG| }t) ||fi |}|j dd }|j+ |j |g|i |S | |j |g|i |S t | t v r*t t | }| |j |g|i |S t- st | d t d| dt. dt0 dt0 ddj3 d t4 D
# t $ r Y w xY w)a
Instantiate one of the video processor classes of the library from a pretrained model vocabulary.
The video 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), or when it's
missing, by falling back to using pattern matching on `pretrained_model_name_or_path`:
List options
Params:
pretrained_model_name_or_path (`str` or `os.PathLike`):
This can be either:
- a string, the *model id* of a pretrained video_processor hosted inside a model repo on
huggingface.co.
- a path to a *directory* containing a video processor file saved using the
[`~video_processing_utils.BaseVideoProcessor.save_pretrained`] method, e.g.,
`./my_model_directory/`.
- a path to a saved video processor JSON *file*, e.g.,
`./my_model_directory/preprocessor_config.json`.
cache_dir (`str` or `os.PathLike`, *optional*):
Path to a directory in which a downloaded pretrained model video processor 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 video processor 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 video processor object. If `True`, then this
functions returns a `Tuple(video_processor, unused_kwargs)` where *unused_kwargs* is a dictionary
consisting of the key/value pairs whose keys are not video processor attributes: i.e., the part of
`kwargs` which has not been used to update `video_processor` 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 video processor attributes will be used to override the
loaded values. Behavior concerning key/value pairs whose keys are *not* video processor 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 AutoVideoProcessor
>>> # Download video processor from huggingface.co and cache.
>>> video_processor = AutoVideoProcessor.from_pretrained("llava-hf/llava-onevision-qwen2-0.5b-ov-hf")
>>> # If video processor files are in a directory (e.g. video processor was saved using *save_pretrained('./test/saved_model/')*)
>>> # video_processor = AutoVideoProcessor.from_pretrained("./test/saved_model/")
```configNtrust_remote_codeT
_from_autovideo_processor_typerY auto_mapimage_processor_typeImageProcessorVideoProcessorAutoImageProcessorz
transformers.z--r
code_revisionzT requires `torchvision` to be installed. Please install `torchvision` and try again.z Unrecognized video processor in z2. Should have a `video_processor_type` key in its z of z3, or one of the following `model_type` keys in its z: z, c 3 K | ] }| y w)N ).0cs r< z5AutoVideoProcessor.from_pretrained.. s @jq@js )popr get_video_processor_dictgetreplacer=
isinstancer r from_pretrainedr2 r7 rd
ValueErrortyper4
__module__
startswithsplitr r register_for_auto_classr r r joinr )clsr> inputsrM r` ra config_dict_video_processor_classvideo_processor_auto_mapimage_processor_class video_processor_class_inferredimage_processor_auto_maphas_remote_codehas_local_codeexplicit_local_code
upstream_repo class_refs r< rt z"AutoVideoProcessor.from_pretrained s L Hd+ "JJ':D A #| +DDEbmflm Q +0F M #' ;??:r#B B'2:'>?S'T $ ! (-E-M$/OO4JD$Q !$01F1N1NO_aq1r. 33QR^,J)#{z2'FF+6z+BCW+X(+C+K+KL\^n+o( ! (
!&*:; ' 7 75 IZ ^d F
)08NPT(U%6:.3G6??3Z/5?S/T, ! ,$CDY$Z !2$>.d:ed6lNe>e , 1 ! J% >t DQ G
$
9!#@.Racp! 09L0I$A)Mj$unt$u !
?D1A ! 9 9 ;8(889VjY_jcij j
"
.8(889VjY_jcij j
&\ 4
4$;DL$I !$0<,<<=Zn]cngmnn () 01 2F G
./L.M N11E0Fd;- X((3}Btyy@jLi@j7j6k
m
U
s A J9 9 KKc 4 t j | || y)a7
Register a new video processor for this class.
Args:
config_class ([`PreTrainedConfig`]):
The configuration corresponding to the model to register.
video_processor_class ([`BaseVideoProcessor`]):
The video processor to register.
)exist_okN)r4 register) config_classr r s r< r zAutoVideoProcessor.register s (( 7LW_(`r^ N)F) r- rw __qualname____doc__r] classmethodr r rt staticmethodr rk r^ r< rY rY sM
&'DEW
F W
r a ar^ rY r4 )NFNNNF)6r r0 os collectionsr typingr configuration_utilsr dynamic_module_utilsr r utilsr r
r r r
r r r utils.import_utilsr video_processing_utilsr auto_factoryr
auto_mappingsr configuration_autor r r r
get_loggerr- rK strtuple__annotations__%MISSING_VIDEO_PROCESSOR_MAPPING_NAMESupdater/
model_typevideo_processorsfast_video_processor_classr4 r= PathLikebooldictrT rY __all__rk r^ r< r s # 4 \ + 8 * 8
H % VaUb!;sE#*cDj:P4Q/Q#Rb -8
-) )!(()NO$G$A$G$G$I K J !1 $ %%)"0J!*-K ++?A^_ 4 +/ %) # "u #u R[[ 4'u u #s(^d
" u
#: u Dj
u u p
,-xa xa .xav %&:
;r^