kernels-community
/

flash-attn2

Model card Files Files and versions

lewtun HF Staff commited on 11 days ago

Commit

4592a27

·

verified ·

1 Parent(s): 35815ed

Fix repo ID in example

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -4,7 +4,7 @@ tags:
   - kernels
 ---
-<!-- ![Status](https://hubwebhook.dholtz.com/shield?repo=kernels-community/flash-attn) -->
 # Flash Attention
@@ -27,7 +27,7 @@ from kernels import get_kernel
 # Setup
 torch.manual_seed(42)
-flash_attn = get_kernel("kernels-community/flash-attn")
 device = torch.device("cuda")
 # Create test tensors

   - kernels
 ---
+<!-- ![Status](https://hubwebhook.dholtz.com/shield?repo=kernels-community/flash-attn2) -->
 # Flash Attention
 # Setup
 torch.manual_seed(42)
+flash_attn = get_kernel("kernels-community/flash-attn2")
 device = torch.device("cuda")
 # Create test tensors