Use pyproject.toml to specify build dependencies

Dao-AILab · Apr 27, 2023 · d63cfc3 · d63cfc3
1 parent ad11394
commit d63cfc3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -51,7 +51,7 @@ pip install flash-attn
 
 Alternatively you can compile from source:
 ```
-python setup.py install
+python -m pip install .
 ```
 
 Interface: `src/flash_attention.py`

diff --git a/pyproject.toml b/pyproject.toml
@@ -0,0 +1,3 @@
+[build-system]
+requires = ["packaging", "setuptools", "torch", "wheel"]
+build-backend = "setuptools.build_meta"