tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

add default torch version if not installed, and support for xformers new wheels (#2049)

Browse Source
This commit is contained in:
Wing Lian
2024-11-13 13:16:47 -05:00
committed by GitHub
parent 010d0e7ff3
commit f2145a3ccb
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
setup.py
View File

@@ -39,7 +39,10 @@ def parse_requirements():
else: else:
# detect the version of torch already installed # detect the version of torch already installed
# and set it so dependencies don't clobber the torch version # and set it so dependencies don't clobber the torch version
torch_version = version("torch") try:
torch_version = version("torch")
except PackageNotFoundError:
torch_version = "2.5.1"
_install_requires.append(f"torch=={torch_version}") _install_requires.append(f"torch=={torch_version}")
version_match = re.match(r"^(\d+)\.(\d+)(?:\.(\d+))?", torch_version) version_match = re.match(r"^(\d+)\.(\d+)(?:\.(\d+))?", torch_version)
@@ -54,6 +57,10 @@ def parse_requirements():
if (major, minor) >= (2, 5): if (major, minor) >= (2, 5):
_install_requires.pop(_install_requires.index(xformers_version)) _install_requires.pop(_install_requires.index(xformers_version))
if patch == 0:
_install_requires.append("xformers==0.0.28.post2")
else:
_install_requires.append("xformers==0.0.28.post3")
_install_requires.pop(_install_requires.index(autoawq_version)) _install_requires.pop(_install_requires.index(autoawq_version))
elif (major, minor) >= (2, 4): elif (major, minor) >= (2, 4):
if patch == 0: if patch == 0: