Fix the bug that torch version less than 1.12 throws TypeError (#1671)

8cecc66a · Chino · GitHub · 35b66c8e · 8cecc66a
Unverified Commit 8cecc66a authored Dec 15, 2022 by Chino Committed by GitHub Dec 14, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/utils/import_utils.py src/diffusers/utils/import_utils.py +1 -1

No files found.
--- a/src/diffusers/utils/import_utils.py
+++ b/src/diffusers/utils/import_utils.py
@@ -204,7 +204,7 @@ try:
    if _torch_available:
        import torch

-        if torch.__version__ < version.Version("1.12"):
+        if version.Version(torch.__version__) < version.Version("1.12"):
            raise ValueError("PyTorch should be >= 1.12")
    logger.debug(f"Successfully imported xformers version {_xformers_version}")
 except importlib_metadata.PackageNotFoundError: