pyproject.toml

[build-system]
requires = ["uv", "setuptools>=61.0", "wheel"] # uv for uv-aware builds, setuptools for packaging
build-backend = "setuptools.build_meta"

[project]
name = "cquantize"
version = "0.1.0"
description = "Quantization script module for confidentialmind-graph project for 4bit GPTQ quantizations (so far)"
readme = "README.md"
requires-python = ">=3.11,<=3.13.10" # 3.13.8 is used in the main project

dependencies = [
    "python-dotenv>=1.0.1",
    "gptqmodel>=1.9.0",
    "threadpoolctl>=3.5.0",
    "tokenicer>=0.0.2",
    "device-smi>=0.3.3",
    "pillow>=11.1.0",
    "torch>=2.6.0",
    "accelerate>=1.3.0",
    "safetensors>=0.5.2",
    "transformers>=4.48.3",
    "datasets>=3.3.0",
    "huggingface-hub>=0.28.1",
    "typer>=0.15.1",
    "hf-transfer>=0.1.9",
]

[tool.setuptools.package-data]
quantize = ["README.md", "*.py"] # Include README and Python files if packaged