first pass at pytorch 2.5.0 support (#1982)

* first pass at pytorch 2.5.0 support

* attempt to install causal_conv1d with mamba

* gracefully handle missing xformers

* fix import

* fix incorrect version, add 2.5.0

* increase tests timeout
This commit is contained in:
Wing Lian
2024-10-21 11:00:45 -04:00
committed by GitHub
parent 67f744dc8c
commit e12a2130e9
12 changed files with 120 additions and 55 deletions

View File

@@ -50,7 +50,9 @@ def parse_requirements():
else:
raise ValueError("Invalid version format")
if (major, minor) >= (2, 4):
if (major, minor) >= (2, 5):
_install_requires.pop(_install_requires.index(xformers_version))
elif (major, minor) >= (2, 4):
if patch == 0:
_install_requires.pop(_install_requires.index(xformers_version))
_install_requires.append("xformers>=0.0.27")
@@ -102,6 +104,7 @@ setup(
],
"mamba-ssm": [
"mamba-ssm==1.2.0.post1",
"causal_conv1d",
],
"auto-gptq": [
"auto-gptq==0.5.1",