diff --git a/src/diffusers/utils/deprecation_utils.py b/src/diffusers/utils/deprecation_utils.py index 78ffdb0a8d..5f276382b6 100644 --- a/src/diffusers/utils/deprecation_utils.py +++ b/src/diffusers/utils/deprecation_utils.py @@ -14,10 +14,10 @@ from packaging import version # "transformers_version": (">=", "5.0.0"), # (operation, version) tuple # } # } -TRANSFORMERS_CLASS_REMAPPING = { +_TRANSFORMERS_CLASS_REMAPPING = { "CLIPFeatureExtractor": { "new_class": "CLIPImageProcessor", - "transformers_version": (">=", "4.57.0"), + "transformers_version": (">", "4.57.0"), }, } @@ -32,12 +32,12 @@ def _should_remap_transformers_class(class_name: str) -> Optional[str]: Returns: The new class name if remapping should occur, None otherwise """ - if class_name not in TRANSFORMERS_CLASS_REMAPPING: + if class_name not in _TRANSFORMERS_CLASS_REMAPPING: return None from .import_utils import is_transformers_version - mapping = TRANSFORMERS_CLASS_REMAPPING[class_name] + mapping = _TRANSFORMERS_CLASS_REMAPPING[class_name] operation, required_version = mapping["transformers_version"] # Only remap if the transformers version meets the requirement