Skip to content

Commit

Permalink
Add how to import FA3 to documentation (Dao-AILab#1112)
Browse files Browse the repository at this point in the history
  • Loading branch information
AdamLouly authored Dec 6, 2024
1 parent b518517 commit df1a744
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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:**
Expand Down

0 comments on commit df1a744

Please sign in to comment.