pyproject.toml

[build-system]
requires = ["setuptools>=68"]
build-backend = "setuptools.build_meta"

[project]
name = "memguard"
version = "0.1.0"
description = "A memorization / training-data-leakage guardrail for diffusion models (anisotropy-of-log-probability)."
readme = "README.md"
requires-python = ">=3.9"
authors = [{ name = "Rohan Asthana" }]
license = { text = "MIT" }
dependencies = [
    "numpy",
    "pillow",
]

[project.optional-dependencies]
api = ["fastapi", "uvicorn[standard]", "python-multipart"]
demo = ["gradio"]
diffusers = ["torch", "diffusers", "transformers", "accelerate"]
dev = ["pytest", "httpx", "ruff"]

[tool.setuptools.packages.find]
include = ["memguard*"]

[tool.ruff]
line-length = 100