diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index cd997e8d057d..0351e3a6d884 100644 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -99,6 +99,7 @@ "file_utils": [], "generation": ["GenerationConfig"], "hf_argparser": ["HfArgumentParser"], + "image_transforms": [], "integrations": [ "is_clearml_available", "is_comet_available", @@ -746,7 +747,6 @@ ] else: _import_structure["image_processing_utils"] = ["ImageProcessingMixin"] - _import_structure["image_transforms"] = ["rescale", "resize", "to_pil_image"] _import_structure["image_utils"] = ["ImageFeatureExtractionMixin"] _import_structure["models.beit"].extend(["BeitFeatureExtractor", "BeitImageProcessor"]) _import_structure["models.bit"].extend(["BitImageProcessor"]) @@ -3944,7 +3944,6 @@ from .utils.dummy_vision_objects import * else: from .image_processing_utils import ImageProcessingMixin - from .image_transforms import rescale, resize, to_pil_image from .image_utils import ImageFeatureExtractionMixin from .models.beit import BeitFeatureExtractor, BeitImageProcessor from .models.bit import BitImageProcessor diff --git a/src/transformers/utils/dummy_vision_objects.py b/src/transformers/utils/dummy_vision_objects.py index 1aa5e29a7d3a..b9e0b80d0d76 100644 --- a/src/transformers/utils/dummy_vision_objects.py +++ b/src/transformers/utils/dummy_vision_objects.py @@ -10,18 +10,6 @@ def __init__(self, *args, **kwargs): requires_backends(self, ["vision"]) -def rescale(*args, **kwargs): - requires_backends(rescale, ["vision"]) - - -def resize(*args, **kwargs): - requires_backends(resize, ["vision"]) - - -def to_pil_image(*args, **kwargs): - requires_backends(to_pil_image, ["vision"]) - - class ImageFeatureExtractionMixin(metaclass=DummyObject): _backends = ["vision"]