[project]
name = "deit-tiny-int8-litert-runtime"
version = "0.1.0"
description = "Runtime dependencies for example.py (DeiT-Tiny INT8, LiteRT + XNNPACK)"
requires-python = ">=3.14,<3.15"
dependencies = [
    "ai-edge-litert==2.2.0",
    "torch==2.13.0",
    "torchvision==0.28.0",
    "pillow==12.3.0",
    "numpy==2.5.2",
]

[tool.uv]
package = false

[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true

[tool.uv.sources]
# The default PyPI aarch64 Linux wheel for torch links against CUDA
# (aarch64 + CUDA is common on NVIDIA Grace/Jetson) and fails to import
# without a CUDA runtime present. This target is a CPU-only Arm server
# with no GPU, so pull the CPU-only build instead.
torch = { index = "pytorch-cpu" }
torchvision = { index = "pytorch-cpu" }

[tool.ai-portal.deployment]
schema-version = "1"
runtime = "litert"
required-capabilities = ["CPU"]

[tool.ai-portal.deployment.ubuntu]
packages = []

[tool.ai-portal.deployment.raspbian]
packages = []

[tool.ai-portal.deployment.files]
expected = ["predictions.json"]