From 06d940efc353fb9138c93a818c940053fae7868a Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Wed, 8 Feb 2023 19:36:20 +0100 Subject: [PATCH] Fixing backward compatiblity `image_processor` in pipeline. (#21513) --- src/transformers/pipelines/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/transformers/pipelines/__init__.py b/src/transformers/pipelines/__init__.py index 239e5c419e..0446064939 100755 --- a/src/transformers/pipelines/__init__.py +++ b/src/transformers/pipelines/__init__.py @@ -861,6 +861,10 @@ def pipeline( image_processor = model_name elif isinstance(config, str): image_processor = config + # Backward compatibility, as `feature_extractor` used to be the name + # for `ImageProcessor`. + elif feature_extractor is not None and isinstance(feature_extractor, BaseImageProcessor): + image_processor = feature_extractor else: # Impossible to guess what is the right image_processor here raise Exception(