Hugging Face
Models
Datasets
Spaces
Buckets
new
Docs
Enterprise
Pricing
Log In
Sign Up
kernels-community
/
flash-attn2
like
31
Follow
kernels-community
403
Kernels
License:
bsd-3-clause
Model card
Files
Files and versions
xet
Community
4
Use this model
b833fce
flash-attn2
15.4 GB
Ctrl+K
Ctrl+K
6 contributors
History:
12 commits
drbh
fix: update library for all ops
b833fce
about 1 year ago
build
fix: update library for all ops
about 1 year ago
flash_attn
fix: revise bindings and wrapper typing
about 1 year ago
tests
feat: include source and enable build
about 1 year ago
torch-ext
fix: revise bindings and wrapper typing
about 1 year ago
.gitattributes
Safe
1.56 kB
feat: pass vars into fwd and include build
about 1 year ago
.gitignore
4 Bytes
feat: include source and enable build
about 1 year ago
build.toml
Safe
5.93 kB
feat: build for all caps 8 and up
about 1 year ago
flake.lock
Safe
3.09 kB
fix: limit build head size and cuda caps for dev
about 1 year ago
flake.nix
Safe
326 Bytes
fix: limit build head size and cuda caps for dev
about 1 year ago
