Fix `is_vision_available` (#24853)

fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

Fix `is_vision_available` (#24853)
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
e4a52b6a · Yih-Dar · GitHub · 4f088870 · e4a52b6a
Unverified Commit e4a52b6a authored Jul 17, 2023 by Yih-Dar Committed by GitHub Jul 17, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

src/transformers/utils/import_utils.py src/transformers/utils/import_utils.py +4 -1

No files found.
--- a/src/transformers/utils/import_utils.py
+++ b/src/transformers/utils/import_utils.py
@@ -546,7 +546,10 @@ def is_vision_available():
        try:
            package_version = importlib.metadata.version("Pillow")
        except importlib.metadata.PackageNotFoundError:
-            return False
+            try:
+                package_version = importlib.metadata.version("Pillow-SIMD")
+            except importlib.metadata.PackageNotFoundError:
+                return False
        logger.debug(f"Detected PIL version {package_version}")
    return _pil_available