From 454ec3d73ced089d25c64bb4bb7e186a7a8cff0c Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 30 Jul 2024 19:57:53 -0700 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 8677879aad..f4e4257b2b 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -203,12 +203,12 @@ elif Version(torch_version) < Version("2.3.0") and Version(xformers_version) >= f"Unsloth: You have torch = {torch_version} but xformers = {xformers_version}.\n"\ f"Please install xformers < 0.0.26 for torch = {torch_version}." ) -elif Version(torch_version) < Version("2.4.0") and Version(xformers_version) >= Version("0.0.27"): - raise ImportError( - f"Unsloth: You have torch = {torch_version} but xformers = {xformers_version}.\n"\ - f"Please install xformers < 0.0.27 for torch = {torch_version}." - ) -pass +# elif Version(torch_version) < Version("2.4.0") and Version(xformers_version) >= Version("0.0.27"): +# raise ImportError( +# f"Unsloth: You have torch = {torch_version} but xformers = {xformers_version}.\n"\ +# f"Please install xformers < 0.0.27 for torch = {torch_version}." +# ) +# pass from xformers._cpp_lib import _register_extensions try: