deit-tiny-int8-litert / pyproject.toml
aorabdel's picture
Sync model repo (text/metadata)
a514d75 verified
Raw History Blame Contribute Delete
1.08 kB
[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"]