Tensor-Pruning / pyproject.toml
Angshul's picture
Upload 14 files
a42d9ad verified
Raw History Blame Contribute Delete
668 Bytes
[build-system]
requires = ["setuptools>=68"]
build-backend = "setuptools.build_meta"
[project]
name = "tensor-pruning"
version = "1.0.0"
description = "Training-free global-budget tensor pruning for LLM query projections"
readme = "README.md"
requires-python = ">=3.10"
dependencies = ["numpy>=1.24"]
[project.optional-dependencies]
http = ["fastapi>=0.115,<1", "uvicorn>=0.30,<1"]
space = ["fastapi>=0.115,<1", "uvicorn>=0.30,<1", "ml-dtypes>=0.4,<1"]
test = ["httpx>=0.27,<1"]
[project.scripts]
tensor-prune = "tensor_pruning.cli:main"
[tool.setuptools.packages.find]
include = ["tensor_pruning*"]
[tool.setuptools.package-data]
tensor_pruning = ["kernel.cpp"]