From df1a744887198e8c3de984244b5f6ceb647cc222 Mon Sep 17 00:00:00 2001 From: Adam Louly Date: Thu, 5 Dec 2024 17:39:18 -0800 Subject: [PATCH] Add how to import FA3 to documentation (#1112) --- README.md | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 9f22ea74b..9cc10d337 100644 --- a/README.md +++ b/README.md @@ -56,8 +56,11 @@ To run the test: export PYTHONPATH=$PWD pytest -q -s test_flash_attn.py ``` - - +Once the package is installed, you can import it as follows: +```python +import flash_attn_interface +flash_attn_interface.flash_attn_func() +``` ## Installation and features **Requirements:**