Skip to content

Commit 34eb665

Browse files
authored
hotfix: fixup get_kernel call in kernels README (#224)
This was not updated because it was added as part of the repo merge.
1 parent 6dfe59c commit 34eb665

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

kernels/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ import torch
3030
from kernels import get_kernel
3131

3232
# Download optimized kernels from the Hugging Face hub
33-
activation = get_kernel("kernels-community/activation")
33+
activation = get_kernel("kernels-community/activation", version=1)
3434

3535
# Random tensor
3636
x = torch.randn((10, 10), dtype=torch.float16, device="cuda")

0 commit comments

Comments
 (0)