File size: 594 Bytes
f8cf4de | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | [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*"]
|