add torch 2.9.0 to ci (#3223)

2025-10-30 18:50:26 -04:00
parent 4b1b4fa6d8
commit 633afffacb
3 changed files with 27 additions and 13 deletions
--- a/setup.py
+++ b/setup.py
@@ -66,6 +66,7 @@ def parse_requirements(extras_require_map):
                extras_require_map.pop("fbgemm-gpu")
                extras_require_map["fbgemm-gpu"] = ["fbgemm-gpu-genai==1.4.1"]
                extras_require_map["vllm"] = ["vllm==0.11.1"]
+                _install_requires.pop(_install_requires.index(xformers_version))
            elif (major, minor) >= (2, 8):
                extras_require_map.pop("fbgemm-gpu")
                extras_require_map["fbgemm-gpu"] = ["fbgemm-gpu-genai==1.3.0"]