mirror of
https://github.com/rushil-thareja/dp-fusion-lib.git
synced 2026-04-24 12:06:23 +02:00
73 lines
2 KiB
TOML
73 lines
2 KiB
TOML
|
[build-system]
|
||
|
|
requires = ["setuptools>=61.0", "wheel"]
|
||
|
|
build-backend = "setuptools.build_meta"
|
||
|
|
|
||
|
|
[project]
|
||
|
|
name = "dp-fusion-lib"
|
||
|
|
version = "0.1.0"
|
||
|
|
description = "Token-Level Differentially Private Inference for Large Language Models"
|
||
|
|
readme = "README.md"
|
||
|
|
license = {file = "LICENSE"}
|
||
|
|
authors = [
|
||
|
|
{name = "Rushil Thareja", email = "rushil.thareja@mbzuai.ac.ae"}
|
||
|
|
]
|
||
|
|
keywords = ["differential-privacy", "llm", "text-generation", "privacy", "nlp", "renyi-divergence"]
|
||
|
|
classifiers = [
|
||
|
|
"Development Status :: 4 - Beta",
|
||
|
|
"Intended Audience :: Developers",
|
||
|
|
"Intended Audience :: Science/Research",
|
||
|
|
"License :: Other/Proprietary License",
|
||
|
|
"Programming Language :: Python :: 3",
|
||
|
|
"Programming Language :: Python :: 3.8",
|
||
|
|
"Programming Language :: Python :: 3.9",
|
||
|
|
"Programming Language :: Python :: 3.10",
|
||
|
|
"Programming Language :: Python :: 3.11",
|
||
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
||
|
|
"Topic :: Security",
|
||
|
|
]
|
||
|
|
requires-python = ">=3.8"
|
||
|
|
|
||
|
|
dependencies = [
|
||
|
|
"torch>=2.0.0",
|
||
|
|
"transformers>=4.25.0",
|
||
|
|
"accelerate>=0.20.0",
|
||
|
|
"requests>=2.25.0",
|
||
|
|
]
|
||
|
|
|
||
|
|
[project.optional-dependencies]
|
||
|
|
dev = [
|
||
|
|
"pytest>=7.0.0",
|
||
|
|
"pytest-cov>=4.0.0",
|
||
|
|
"black>=23.0.0",
|
||
|
|
"ruff>=0.1.0",
|
||
|
|
"pre-commit>=3.0.0",
|
||
|
|
]
|
||
|
|
docs = [
|
||
|
|
"mkdocs>=1.5.0",
|
||
|
|
"mkdocs-material>=9.0.0",
|
||
|
|
"mkdocstrings[python]>=0.24.0",
|
||
|
|
]
|
||
|
|
|
||
|
|
[project.urls]
|
||
|
|
Homepage = "https://github.com/rushil-thareja/dp-fusion-lib"
|
||
|
|
Documentation = "https://rushil-thareja.github.io/dp-fusion-lib"
|
||
|
|
Repository = "https://github.com/rushil-thareja/dp-fusion-lib"
|
||
|
|
Issues = "https://github.com/rushil-thareja/dp-fusion-lib/issues"
|
||
|
|
Paper = "https://arxiv.org/abs/2507.04531"
|
||
|
|
|
||
|
|
[tool.setuptools.packages.find]
|
||
|
|
where = ["src"]
|
||
|
|
|
||
|
|
[tool.black]
|
||
|
|
line-length = 100
|
||
|
|
target-version = ["py38", "py39", "py310", "py311"]
|
||
|
|
|
||
|
|
[tool.ruff]
|
||
|
|
line-length = 100
|
||
|
|
select = ["E", "F", "I", "N", "W"]
|
||
|
|
|
||
|
|
[tool.pytest.ini_options]
|
||
|
|
testpaths = ["tests"]
|
||
|
|
python_files = ["test_*.py"]
|
||
|
|
addopts = "-v --cov=dp_fusion_lib --cov-report=term-missing"
|