fix: refactor bindings and update py library for correctness 876ac68
drbh commited on
How to use replicate/flash-attn2 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/flash-attn2", version=1)