pyproject.toml

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

[project]
name = "freeflowx"
version = "0.1.0"
description = "Small-scale data-free flow-map distillation research harness"
requires-python = ">=3.10"
dependencies = [
  "torch",
  "torchvision",
  "numpy",
  "scipy",
  "pandas",
  "matplotlib",
  "tqdm",
  "pillow",
  "pyyaml",
  "wandb>=0.16",
  "torchmetrics[image]>=1.3",
  "torch-fidelity>=0.3",
  "clean-fid>=0.1.35",
  "einops>=0.7",
  "timm>=0.9",
  "datasets>=2.19",
  "huggingface_hub>=0.23",
  "diffusers>=0.30",
  "accelerate>=0.31",
  "transformers>=4.41",
  "safetensors>=0.4",
]

[tool.setuptools.packages.find]
where = ["."]
include = ["freeflowx*"]