riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-16 11:08:23 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

[video processor] fix BC when no video config if found (#38840)

Browse Source 
fix auto video processor
This commit is contained in:
Raushan Turganbay 2025-06-17 09:20:16 +02:00 committed by GitHub
parent e61160c5db
commit bf370e446b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/transformers/models/auto/video_processing_auto.py
View File

@ -306,15 +306,15 @@ class AutoVideoProcessor:
if "AutoVideoProcessor" in config_dict.get("auto_map", {}): if "AutoVideoProcessor" in config_dict.get("auto_map", {}):
video_processor_auto_map = config_dict["auto_map"]["AutoVideoProcessor"] video_processor_auto_map = config_dict["auto_map"]["AutoVideoProcessor"]
# If we still don't have the video processor class, check if we're loading from a previous feature extractor config # If we still don't have the video processor class, check if we're loading from a previous image processor config
# and if so, infer the video processor class from there. # and if so, infer the video processor class from there.
if video_processor_class is None and video_processor_auto_map is None: if video_processor_class is None and video_processor_auto_map is None:
feature_extractor_class = config_dict.pop("feature_extractor_type", None) image_processor_class = config_dict.pop("image_processor_type", None)
if feature_extractor_class is not None: if image_processor_class is not None:
video_processor_class = feature_extractor_class.replace("FeatureExtractor", "VideoProcessor") video_processor_class = image_processor_class.replace("ImageProcessor", "VideoProcessor")
if "AutoFeatureExtractor" in config_dict.get("auto_map", {}): if "AutoImageProcessor" in config_dict.get("auto_map", {}):
feature_extractor_auto_map = config_dict["auto_map"]["AutoFeatureExtractor"] image_processor_auto_map = config_dict["auto_map"]["AutoImageProcessor"]
video_processor_auto_map = feature_extractor_auto_map.replace("FeatureExtractor", "VideoProcessor") video_processor_auto_map = image_processor_auto_map.replace("ImageProcessor", "VideoProcessor")
# If we don't find the video processor class in the video processor config, let's try the model config. # If we don't find the video processor class in the video processor config, let's try the model config.
if video_processor_class is None and video_processor_auto_map is None: if video_processor_class is None and video_processor_auto_map is None: