squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #193 from anthonyhu/pyproject-build

Browse Source 
Use pyproject.toml to specify build dependencies
This commit is contained in:
Tri Dao 2023-05-11 21:26:28 -04:00 committed by GitHub
commit 36d0a19f1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -51,7 +51,7 @@ pip install flash-attn
Alternatively you can compile from source: Alternatively you can compile from source:
``` ```
python setup.py install python -m pip install .
``` ```
Interface: `src/flash_attention.py` Interface: `src/flash_attention.py`

3
pyproject.toml Normal file
View File

@ -0,0 +1,3 @@
[build-system]
requires = ["packaging", "setuptools", "torch", "wheel"]
build-backend = "setuptools.build_meta"