Instructions to use replicate/flash-attn2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Kernels
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) - Notebooks
- Google Colab
- Kaggle
Commit History
-
fix: refactor bindings and update py library for correctness
drbh876ac68
-
feat improve readme and library code
drbhd6cc1b0
-
feat: pass vars into fwd and include build
drbh39b4aba -
feat: include source and enable build
drbha7165c8