| # SPDX-License-Identifier: Apache-2.0 | |
| # Python package of meteor-p150, from the ROOT of the model repo: | |
| # pip install -e . # the Python API | |
| # pip install -e ".[server,test]" # + the HTTP server and the tests | |
| # | |
| # Install it into an environment that already has tt-metal's ttnn (and its torch 2.11.0+cpu): ttnn and torch are | |
| # NOT declared, so pip never replaces the tt-metal build's own torch. The container image installs the same set | |
| # through tt-model.yaml runtime.packages. | |
| # | |
| # This file lives at the repo root ON PURPOSE and must never move into code/: tt-model copies code/ over /opt/tt-metal | |
| # right before `uv pip install -e /opt/tt-metal` in the image build, so a code/pyproject.toml would replace tt-metal's | |
| # own (project ttnn) and break `import ttnn` in the image (research/PACKAGING_PILOT.md problem 1; | |
| # research/packaging/scripts/check_bundle.py refuses it). It is not in tt-model.yaml extra_code: it reaches the Hub as | |
| # an overlay file next to README.md (SERVING.md section 2), like the other top-level docs. | |
| [build-system] | |
| requires = ["setuptools>=77"] | |
| build-backend = "setuptools.build_meta" | |
| [project] | |
| name = "tt-meteor" | |
| dynamic = ["version"] # tt_meteor.__version__, the version /openapi.json reports | |
| description = "METEOR (TIER IV, AutowareFoundation/meteor) on Tenstorrent Blackhole p150 (tt-nn), with a Python API and an HTTP server" | |
| readme = "code/PYTHON.md" | |
| license = "Apache-2.0" | |
| requires-python = ">=3.10" | |
| dependencies = [ | |
| "numpy>=1.24.4,<2", # ttnn needs numpy 1.x | |
| "pillow", | |
| "pyyaml", # meteor_v157.param.yaml | |
| "huggingface_hub", # weights pointer: AutowareFoundation/meteor@01a5f6d71df5ecbbb5853ec600825481d57b9c6b | |
| "safetensors", | |
| "onnx>=1.17,<2", # reads the ONNX initializers (data only) | |
| "opencv-python-headless>=4.8,<4.12", # host post-processing (seg-fusion warp, rotated NMS); tt-metal's env has 4.8.1 | |
| ] | |
| [project.optional-dependencies] | |
| server = ["fastapi", "uvicorn", "pydantic>=2"] | |
| test = ["pytest", "httpx"] | |
| reference = ["onnxruntime"] # ONNX Runtime golden (research / tests only, not in the image) | |
| [project.urls] | |
| "Model card" = "https://huggingface.co/changh95/meteor-p150" | |
| "Weights" = "https://huggingface.co/AutowareFoundation/meteor" | |
| "METEOR runtime (no Autoware package)" = "https://github.com/tier4/METEOR/tree/dc193a81e9959de57b751e269127ae9b67e3f300/deploy/cpp" | |
| [tool.setuptools.dynamic] | |
| version = { attr = "tt_meteor.__version__" } | |
| [tool.setuptools.packages.find] | |
| where = ["code"] # import root (code/conftest.py and code/scripts/ are not packaged) | |
| include = ["tt_meteor*"] # with the vendored tt_meteor.ttaw (common/tools/vendor.py; never edit it) | |
| exclude = ["tt_meteor.tests*"] | |
| [tool.setuptools.package-data] | |
| "tt_meteor" = ["samples/*", "samples/*/*", "calib/*.json"] | |
| "tt_meteor.tt" = ["kernels/*.cpp", "kernels/*.hpp", "kernels/*.h"] | |
| "tt_meteor.ttaw" = ["API.md", "VENDORED.json"] | |
| "tt_meteor.ttaw.ops" = ["kernels/*.cpp", "kernels/*.hpp", "kernels/*.h"] | |
| [tool.pytest.ini_options] | |
| markers = ["device: needs a Tenstorrent chip (skipped on host-only runs)"] | |