JahJedi commited on
Commit
8109741
Β·
verified Β·
1 Parent(s): 7b00d99

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ flash_attn-2.8.3.post1-cp312-cp312-linux_x86_64.whl filter=lfs diff=lfs merge=lfs -text
37
+ sageattention-2.2.0-cp312-cp312-linux_x86_64.whl filter=lfs diff=lfs merge=lfs -text
LICENSE ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ This repository redistributes binary wheels of two open-source projects.
2
+
3
+ - SageAttention 2.2 β€” licensed under Apache License 2.0
4
+ https://github.com/thu-ml/SageAttention/blob/main/LICENSE
5
+ - FlashAttention 2.8.3 β€” licensed under BSD 3-Clause License
6
+ https://github.com/Dao-AILab/flash-attention/blob/main/LICENSE
7
+
8
+ Each wheel contains its own LICENSE file inside its dist-info/ directory.
9
+ Redistribution of these wheels is permitted under both licenses; users
10
+ are responsible for complying with the respective terms.
README.md ADDED
@@ -0,0 +1,150 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - sageattention
5
+ - flash-attention
6
+ - flash-attn
7
+ - blackwell
8
+ - sm_120
9
+ - rtx-pro-6000
10
+ - rtx-5090
11
+ - cuda-13
12
+ - pytorch-2.11
13
+ - python-3.12
14
+ - wheels
15
+ - prebuilt
16
+ - comfyui
17
+ language:
18
+ - en
19
+ ---
20
+
21
+ # SageAttention 2.2 + FlashAttention 2.8.3 β€” Linux wheels for Blackwell (sm_120) on PyTorch 2.11 / CUDA 13.0
22
+
23
+ Community-built `.whl` files for a combo that, as of June 2026, isn't shipped by upstream:
24
+
25
+ - **SageAttention 2.2.0** β€” only Windows wheels in upstream releases; Linux requires source build
26
+ - **FlashAttention 2.8.3.post1** β€” Dao-AILab releases don't cover cu130 + torch 2.11 yet
27
+
28
+ If your hardware/software matches the table below, these install in seconds instead of 5–40 minutes of source compilation.
29
+
30
+ ---
31
+
32
+ ## Compatibility
33
+
34
+ | Requirement | Value | Why it matters |
35
+ |---|---|---|
36
+ | Python | **3.12.x** (built on 3.12.3) | `cp312` ABI tag β€” won't load on 3.11 or 3.13 |
37
+ | PyTorch | **2.11.0+cu130** | C++ ABI must match β€” won't load with `cu126` torch |
38
+ | CUDA toolkit | 13.0 (driver compatible with cu130) | required for runtime |
39
+ | GPU compute capability | **12.0 (Blackwell)** β€” RTX PRO 6000 Blackwell, RTX 5090/5080/5070 Ti, B100/B200, etc. | wheels compiled with `arch=compute_120,code=sm_120` natively (no PTX-JIT fallback) |
40
+ | OS | Linux x86_64 | `manylinux` not applied β€” built on Ubuntu, untested elsewhere |
41
+
42
+ If **any** row doesn't match β€” these wheels probably won't load and you'll need to build from source.
43
+
44
+ ---
45
+
46
+ ## Files & hashes
47
+
48
+ | File | Size | SHA256 |
49
+ |---|---|---|
50
+ | `sageattention-2.2.0-cp312-cp312-linux_x86_64.whl` | 14.8 MB | `4a47a462059bf09b7d10be68f93a0a462dd6184068298fc6ff10e94751fa6dd4` |
51
+ | `flash_attn-2.8.3.post1-cp312-cp312-linux_x86_64.whl` | 65.8 MB | `a77fe499d1c7244e0549d9d2700def0c1c82a1b629d993204e9be78debd96cb8` |
52
+
53
+ Verify before installing:
54
+ ```bash
55
+ sha256sum sageattention-2.2.0-cp312-cp312-linux_x86_64.whl
56
+ sha256sum flash_attn-2.8.3.post1-cp312-cp312-linux_x86_64.whl
57
+ ```
58
+
59
+ ---
60
+
61
+ ## Install
62
+
63
+ ```bash
64
+ pip install sageattention-2.2.0-cp312-cp312-linux_x86_64.whl
65
+ pip install flash_attn-2.8.3.post1-cp312-cp312-linux_x86_64.whl
66
+ ```
67
+
68
+ Smoke test:
69
+ ```python
70
+ import torch
71
+ import sageattention
72
+ from sageattention import sageattn
73
+ from flash_attn import flash_attn_func
74
+
75
+ q = torch.randn(2, 8, 256, 64, dtype=torch.float16, device='cuda')
76
+ k = torch.randn(2, 8, 256, 64, dtype=torch.float16, device='cuda')
77
+ v = torch.randn(2, 8, 256, 64, dtype=torch.float16, device='cuda')
78
+
79
+ # sageattention expects (B, H, S, D) by default with tensor_layout='HND'
80
+ out_sage = sageattn(q, k, v, tensor_layout='HND', is_causal=False)
81
+
82
+ # flash_attn_func expects (B, S, H, D)
83
+ q2 = q.transpose(1, 2).contiguous()
84
+ k2 = k.transpose(1, 2).contiguous()
85
+ v2 = v.transpose(1, 2).contiguous()
86
+ out_flash = flash_attn_func(q2, k2, v2, causal=False)
87
+
88
+ print('sageattention:', out_sage.shape, 'β€” OK')
89
+ print('flash_attn: ', out_flash.shape, 'β€” OK')
90
+ ```
91
+
92
+ ---
93
+
94
+ ## Build environment (exactly what produced these wheels)
95
+
96
+ ```
97
+ OS: Ubuntu 24.04 (kernel 6.14.0-1010-oem)
98
+ Python: 3.12.3
99
+ PyTorch: 2.11.0+cu130
100
+ CUDA: 13.0.r13.0 / V13.0.88 (apt cuda-toolkit-13-0)
101
+ ninja: 1.11.1.4
102
+ setuptools: 81.0.0
103
+ GPU: NVIDIA RTX PRO 6000 Blackwell Workstation Edition (sm_120)
104
+ ```
105
+
106
+ Build flags:
107
+
108
+ **sageattention:**
109
+ ```bash
110
+ TORCH_CUDA_ARCH_LIST="12.0" MAX_JOBS=4 NVCC_THREADS=2 \
111
+ pip install . --no-build-isolation
112
+ ```
113
+
114
+ **flash-attn:**
115
+ ```bash
116
+ TORCH_CUDA_ARCH_LIST="12.0" FLASH_ATTN_CUDA_ARCHS="120" MAX_JOBS=2 NVCC_THREADS=1 \
117
+ pip install flash-attn==2.8.3.post1 --no-build-isolation
118
+ ```
119
+
120
+ Backward kernels are **included** in flash-attn β€” training works, not just inference.
121
+
122
+ ---
123
+
124
+ ## Licenses
125
+
126
+ Both upstream projects allow redistribution; redistributing these binary builds preserves their license terms:
127
+
128
+ - **SageAttention** β€” [Apache License 2.0](https://github.com/thu-ml/SageAttention/blob/main/LICENSE)
129
+ - **FlashAttention** β€” [BSD 3-Clause](https://github.com/Dao-AILab/flash-attention/blob/main/LICENSE)
130
+
131
+ The wheels contain copies of `LICENSE` files in their `*.dist-info/licenses/` directories. Originals are at:
132
+ - https://github.com/thu-ml/SageAttention
133
+ - https://github.com/Dao-AILab/flash-attention
134
+
135
+ ---
136
+
137
+ ## Disclaimers (read these)
138
+
139
+ - **Not official.** This is a community build. Authors of upstream packages have no involvement.
140
+ - **Smoke-tested only.** Each wheel was verified to: (1) import without errors, (2) execute a small forward pass on a Blackwell GPU and return a tensor of expected shape and dtype. No accuracy benchmarks, no model-level integration tests were run.
141
+ - **No warranty.** If these wheels misbehave on your system β€” wrong results, crashes, performance regressions β€” that's on you to verify against a source build before relying on them.
142
+ - **Reproducible-ish.** Wheel SHA-256 may differ slightly between rebuilds due to embedded timestamps even when binary kernel content is identical.
143
+
144
+ ---
145
+
146
+ ## Attribution
147
+
148
+ These wheels were built by **JahJedi** with assistance from **Claude Code** (Anthropic) β€” the diagnosis of an earlier system freeze under unbounded `ninja`, the choice of safe build parameters (`MAX_JOBS`, `NVCC_THREADS`, single-arch targeting), and this README itself were Claude-assisted. Verification commands were run by the human; final responsibility for the published artifacts is the human's.
149
+
150
+ If anything in these wheels turns out to be wrong, ping the human first β€” they can't claim AI ignorance.
flash_attn-2.8.3.post1-cp312-cp312-linux_x86_64.whl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a77fe499d1c7244e0549d9d2700def0c1c82a1b629d993204e9be78debd96cb8
3
+ size 65808081
sageattention-2.2.0-cp312-cp312-linux_x86_64.whl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a47a462059bf09b7d10be68f93a0a462dd6184068298fc6ff10e94751fa6dd4
3
+ size 14856977