diff --git a/pyproject.toml b/pyproject.toml index 2bc890a..1433aca 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,3 +1,19 @@ +[build-system] +requires = ["setuptools", "wheel"] +build-backend = "setuptools.build_meta" + +[project] +name = "stripedhyena" +version = "0.2.0" +description = "Model and inference code for beyond Transformer architectures" +readme = "README.md" +authors = [{ name = "Michael Poli"}] +dependencies = [ + "transformers", + "tokenizers", + "flash_attn>=2.0.0" +] + [tool.black] line-length = 119