| [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"] | |