| [build-system] |
| requires = ["setuptools>=68", "wheel"] |
| build-backend = "setuptools.build_meta" |
|
|
| [project] |
| name = "layer-interactions" |
| version = "0.1.0" |
| description = "Second-order interaction pruning for Hugging Face causal language models" |
| readme = "README.md" |
| requires-python = ">=3.10" |
| authors = [ |
| {name = "Angshul Majumdar"} |
| ] |
| dependencies = [ |
| "torch>=2.2", |
| "transformers>=4.45", |
| "accelerate>=1.0", |
| "psutil>=5.9" |
| ] |
|
|
| [project.optional-dependencies] |
| examples = ["datasets>=3.0"] |
| test = ["pytest>=8.0"] |
|
|
| [tool.setuptools.packages.find] |
| where = ["."] |
| include = ["layer_interactions*"] |
|
|
