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

[project]
name = "jolt-model-runtime"
version = "0.1.0"
description = "Inference runtime for Jolt decision models"
requires-python = ">=3.12"
license = { file = "LICENSE-CODE" }
dependencies = [
  "torch>=2.9",
  "torchvision>=0.24",
  "transformers>=5.17,<5.18",
  "peft>=0.21,<0.22",
  "Pillow>=11",
  "huggingface-hub>=1.0,<2",
  "safetensors>=0.6",
  "flash-linear-attention[cuda]==0.5.2",
]

[tool.setuptools]
package-dir = {"" = "src"}

[tool.setuptools.packages.find]
where = ["src"]