How to use replicate/triton-flash-attn-sink with Kernels:
# !pip install kernels from kernels import get_kernel # a version (or an explicit revision) is required; see the "Files and versions" tab for the available ones kernel = get_kernel("replicate/triton-flash-attn-sink", version=1)
