| # tt-moge: MoGe-2 on Tenstorrent Blackhole, installable on top of an environment that already has ttnn / tt-metal | |
| # (and torch, which comes with ttnn). Only the model's own Python dependencies are declared here. | |
| # | |
| # pip install -e code/ # editable (from the repo root) | |
| # pip install code/ # regular install | |
| # pip install -e "code/[server]" # + the HTTP server's dependencies | |
| # | |
| # NOTE: tt-model's container build copies code/ over /opt/tt-metal, so this file is NOT listed in | |
| # tt-model.yaml source.extra_code (it would replace tt-metal's own pyproject.toml in the image). In the image, | |
| # tt_moge is importable without installing (the tt-metal root is on sys.path). | |
| [build-system] | |
| requires = ["setuptools>=64"] | |
| build-backend = "setuptools.build_meta" | |
| [project] | |
| name = "tt-moge" | |
| version = "0.2.0" | |
| description = "MoGe-2 monocular metric geometry (depth, point map, normals, intrinsics) on one Tenstorrent Blackhole chip" | |
| readme = { text = "See PYTHON.md in https://huggingface.co/changh95/moge-2-p150", content-type = "text/markdown" } | |
| requires-python = ">=3.10" | |
| license = { text = "Apache-2.0 (port); MIT (vendored microsoft/MoGe and utils3d under tt_moge/reference)" } | |
| keywords = ["tenstorrent", "blackhole", "ttnn", "moge", "depth-estimation", "monocular-geometry", "surface-normals"] | |
| dependencies = [ | |
| # ttnn needs numpy < 2 | |
| "numpy>=1.26,<2", | |
| "pillow", | |
| "scipy", | |
| "huggingface_hub", | |
| # module-level import of the vendored moge reference (bounded below the numpy-2-only releases) | |
| "opencv-python-headless<4.12", | |
| ] | |
| [project.optional-dependencies] | |
| server = ["fastapi", "uvicorn", "pydantic>=2", "isal"] | |
| test = ["pytest"] | |
| [project.urls] | |
| "Model card" = "https://huggingface.co/changh95/moge-2-p150" | |
| "Upstream" = "https://github.com/microsoft/MoGe" | |
| "Port" = "https://github.com/changh95/tt-MoGe" | |
| [tool.setuptools.packages.find] | |
| where = ["."] | |
| include = ["tt_moge*"] | |
| namespaces = false | |
| [tool.setuptools.package-data] | |
| # device kernels and the C++ host kernels are data of the tt_moge package (no __init__.py in those dirs); | |
| # tt_moge/tools_opt (optimization scripts, not a package) is not installed | |
| tt_moge = ["kernels/*.cpp", "kernels/*.hpp", "host_ext/*.cpp"] | |
| "*" = ["*.pyi"] | |