pyproject.toml

[project]
name = "py-img-gen"
version = "0.1.0"
description = 'A library for the book "Image Generation with Python"'
authors = [
    { name = "Shunsuke Kitada", email = "shunsuke.kitada.0831@gmail.com" },
    { name = "Image Generation with Python", email = "python.image.generation@gmail.com" },
]
readme = "README.md"
license = "MIT"
keywords = [
    "python",
    "image-generation",
    "diffusers",
    "transformers",
    "torch",
    "torchvision",
    "controlnet",
    "lora",
    "clip",
    "stable-diffusion",
]
classifiers = [
    # Audience
    "Intended Audience :: Developers",
    # Supported Python versions
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
]
requires-python = ">=3.10"
dependencies = [
    "diffusers[torch]>=0.31.0",
    "einops>=0.7.0",
    "matplotlib>=3.5.0",
    "torch>=1.0.0",
    "torchvision>=0.2.1",
    "transformers[torch]>=4.30.0",
]

[project.optional-dependencies]
docs = [
    "furo>=2024.8.6",
    "myst-parser>=4.0.0",
    "sphinx>=8.1.3",
    "sphinx-autodoc-typehints>=3.0.0",
    "sphinx-pyproject>=0.3.0",
]
clip = ["more-itertools>=10.6.0", "scikit-image>=0.25.1"]
sd3 = ["protobuf<3.21.0", "sentencepiece>=0.2.0"]
controlnet = [
    "controlnet-aux>=0.0.9",
    "mediapipe>=0.10.11",
    "opencv-python>=4.10.0.84",
]
quantization = ["bitsandbytes>=0.44.1"]
lora = ["datasets>=3.1.0", "peft>=0.13.2"]
ncsn = ["diffusers-ncsn>=0.1.0"]

[dependency-groups]
dev = ["mypy>=1.0.0", "pytest-mock>=3.14.0", "pytest>=6.0.0", "ruff>=0.1.5"]

[project.urls]
Repository = "https://github.com/py-img-gen/py-img-gen-lib"
Issues = "https://github.com/py-img-gen/py-img-gen-lib/issues"

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.mypy]
python_version = "3.10"
ignore_missing_imports = true