Upload folder using huggingface_hub
Browse files- .gitattributes +2 -0
- LICENSE +10 -0
- README.md +150 -0
- flash_attn-2.8.3.post1-cp312-cp312-linux_x86_64.whl +3 -0
- sageattention-2.2.0-cp312-cp312-linux_x86_64.whl +3 -0
.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
|