| [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 | |