File size: 1,643 Bytes
6ffd3f8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | # Python package of the SuperPoint tt-nn port: pip install -e code/ (or pip install code/)
# Install it into an environment that already has ttnn / tt-metal (ttnn is not a PyPI dependency here).
[build-system]
requires = ["setuptools>=64"]
build-backend = "setuptools.build_meta"
[project]
name = "tt-superpoint"
version = "0.1.0"
description = "SuperPoint keypoint detector and descriptor on a Tenstorrent Blackhole chip (tt-nn)"
readme = "PYTHON.md"
requires-python = ">=3.10"
license = { text = "Apache-2.0 (code); weights: Magic Leap SuperPoint licence, noncommercial research only" }
dependencies = [
"numpy",
"torch",
"pillow",
"transformers>=4.40",
"huggingface_hub",
"safetensors",
]
[project.optional-dependencies]
server = ["fastapi", "uvicorn", "pydantic"]
test = ["pytest"]
[project.urls]
"Model card" = "https://huggingface.co/changh95/superpoint-p150"
"Port source" = "https://github.com/changh95/tt-superpoint"
[tool.setuptools]
# tt_superpoint public API (code/tt_superpoint)
# tt_superpoint._port the port package code/models (the HTTP server keeps importing it as `models`)
# tt_superpoint.kernels the custom device kernel sources (code/kernels), compiled at run time
package-dir = { "tt_superpoint" = "tt_superpoint", "tt_superpoint._port" = "models", "tt_superpoint.kernels" = "kernels" }
packages = [
"tt_superpoint",
"tt_superpoint._port",
"tt_superpoint._port.reference",
"tt_superpoint._port.tt",
"tt_superpoint.kernels",
]
include-package-data = false
[tool.setuptools.package-data]
"tt_superpoint.kernels" = ["**/*.cpp", "**/*.hpp", "**/*.h", "**/*.inc"]
|