Duplicated from SecondNatureComputing/flash-attn-4-sm120
How to use vonkaiser/flash-attn-4-sm120 with Kernels:
# !pip install kernels from kernels import get_kernel kernel = get_kernel("vonkaiser/flash-attn-4-sm120")
