From 8cecc66a74a66451279f4f495b7a9451eda95253 Mon Sep 17 00:00:00 2001 From: Chino Date: Thu, 15 Dec 2022 04:29:39 +0800 Subject: [PATCH] Fix the bug that torch version less than 1.12 throws TypeError (#1671) --- src/diffusers/utils/import_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index ade8ace072..ad3ab69f66 100644 --- 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: