ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update _utils.py

Browse source
This commit is contained in:
Daniel Han 2024-07-30 19:57:53 -07:00
commit b7bdb18552
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
unsloth/models/_utils.py
View file

@ -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: