riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Remove image_transforms functions from init (#20704)

Browse Source
This commit is contained in:
amyeroberts 2022-12-14 10:17:11 +00:00 committed by GitHub
parent d994473b05
commit 9bafedc0fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/__init__.py
View File

@ -99,6 +99,7 @@ _import_structure = {
"file_utils": [],
"generation": ["GenerationConfig"],
"hf_argparser": ["HfArgumentParser"],
"image_transforms": [],
"integrations": [
"is_clearml_available",
"is_comet_available",
@ -748,7 +749,6 @@ except OptionalDependencyNotAvailable:
]
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"])
@ -3949,7 +3949,6 @@ if TYPE_CHECKING:
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

12
src/transformers/utils/dummy_vision_objects.py
View File

@ -10,18 +10,6 @@ class ImageProcessingMixin(metaclass=DummyObject):
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"]