File size: 690 Bytes
c675b8d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | [build-system]
requires = ["setuptools>=61"]
build-backend = "setuptools.build_meta"
[project]
name = "shapequant"
version = "0.1.0"
description = "Data-free, training-free noise-shaped quantization of Hugging Face causal LMs (GPTQ-format output)"
readme = "README.md"
requires-python = ">=3.9"
license = { text = "Apache-2.0" }
authors = [{ name = "Angshul Majumdar", email = "angshul@iiitd.ac.in" }]
dependencies = ["torch>=2.1", "safetensors>=0.4", "transformers>=4.40", "accelerate>=0.26", "huggingface_hub>=0.20"]
[project.optional-dependencies]
eval = ["datasets"]
test = ["pytest"]
[project.scripts]
shapequant = "shapequant.cli:main"
[tool.setuptools]
packages = ["shapequant"]
|