From 4ff96a25263b2d4cadfa9fc81963110d8a6916d3 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 9 Jul 2025 08:43:40 -0400 Subject: [PATCH] fix xformers version (#2888) --- .github/workflows/tests.yml | 4 ++-- setup.py | 5 ++++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 36a90e274..f98d2bedd 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -52,7 +52,7 @@ jobs: fail-fast: false matrix: python_version: ["3.11"] - pytorch_version: ["2.5.1", "2.6.0", "2.7.1"] + pytorch_version: ["2.5.1", "2.6.0", "2.7.0", "2.7.1"] timeout-minutes: 20 steps: @@ -125,7 +125,7 @@ jobs: fail-fast: false matrix: python_version: ["3.11"] - pytorch_version: ["2.5.1", "2.6.0", "2.7.1"] + pytorch_version: ["2.5.1", "2.6.0", "2.7.0", "2.7.1"] timeout-minutes: 20 steps: diff --git a/setup.py b/setup.py index 1b29cd44d..731fe8a6f 100644 --- a/setup.py +++ b/setup.py @@ -66,7 +66,10 @@ def parse_requirements(extras_require_map): if (major, minor) >= (2, 7): _install_requires.pop(_install_requires.index(xformers_version)) - _install_requires.append("xformers==0.0.31.post1") + if patch == 0: + _install_requires.append("xformers==0.0.30") + else: + _install_requires.append("xformers==0.0.31.post1") extras_require_map["vllm"] = ["vllm>=0.9.0"] elif (major, minor) >= (2, 6): _install_requires.pop(_install_requires.index(xformers_version))