Skip to content

Commit a2d8d81

Browse files
committed
feat: use artifacts to style results page
1 parent 1bf51ce commit a2d8d81

File tree

2 files changed

+562
-23
lines changed

2 files changed

+562
-23
lines changed

flash_attn/impls/hf_kernels_flash_attn.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ import os
2121
import kernels_benchmark_tools as kbt
2222
from kernels import get_kernel
2323

24-
hf_kernels_flash_attn = get_kernel("kernels-community/flash-attn")
24+
hf_kernels_flash_attn = get_kernel("kernels-community/flash-attn", revision="v0.0.2")
2525

2626

2727
def hf_flash_attention(query, key, value):

0 commit comments

Comments
 (0)