diff --git a/src/transformers/models/auto/video_processing_auto.py b/src/transformers/models/auto/video_processing_auto.py index d130230345d1..d8bcc7814667 100644 --- a/src/transformers/models/auto/video_processing_auto.py +++ b/src/transformers/models/auto/video_processing_auto.py @@ -91,7 +91,7 @@ def video_processor_class_from_name(class_name: str): for module_name, extractors in VIDEO_PROCESSOR_MAPPING_NAMES.items(): - if class_name in extractors: + if extractors is not None and class_name in extractors: module_name = model_type_to_module_name(module_name) module = importlib.import_module(f".{module_name}", "transformers.models")