Skip to content

Commit 7f0d4c3

Browse files
committed
Add post-fix xformers version for proper cuda compatibility
Signed-off-by: Mustafa Eyceoz <[email protected]>
1 parent 58c87a3 commit 7f0d4c3

File tree

1 file changed

+1
-16
lines changed

1 file changed

+1
-16
lines changed

pyproject.toml

Lines changed: 1 addition & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -65,22 +65,7 @@ cuda = [
6565
lora = [
6666
"unsloth>=2025.10.11",
6767
"trl>=0.18.0",
68-
# PyTorch's optimized xformers for CUDA 12.8 (compatible with CUDA 12.4+)
69-
"xformers @ https://download.pytorch.org/whl/cu128/xformers-0.0.33%2B5d4b92a5.d20251029-cp39-abi3-linux_x86_64.whl",
70-
]
71-
72-
lora-cu129 = [
73-
"unsloth>=2025.10.11",
74-
"trl>=0.18.0",
75-
# PyTorch's optimized xformers for CUDA 12.9
76-
"xformers @ https://download.pytorch.org/whl/cu129/xformers-0.0.33%2B5d4b92a5.d20251029-cp39-abi3-linux_x86_64.whl",
77-
]
78-
79-
lora-cu130 = [
80-
"unsloth>=2025.10.11",
81-
"trl>=0.18.0",
82-
# PyTorch's optimized xformers for CUDA 13.0+
83-
"xformers @ https://download.pytorch.org/whl/cu130/xformers-0.0.33%2B5d4b92a5.d20251029-cp39-abi3-linux_x86_64.whl",
68+
"xformers>=0.0.33.post1",
8469
]
8570

8671
dev = [

0 commit comments

Comments
 (0)