File size: 1,329 Bytes
1b17257 77e087f 1b17257 | 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 | [build-system]
requires = ["setuptools>=64"]
build-backend = "setuptools.build_meta"
[project]
name = "hamer-p150"
version = "0.1.0"
description = "HaMeR hand mesh recovery (ViT-H/16 + MANO head) on a Tenstorrent Blackhole p150 via tt-nn"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "MIT code; MANO model non-commercial (see the model card)" }
# ttnn / tt-metal is NOT declared: install this package into an environment that already has a built tt-metal
# (its python_env or a venv with ttnn). Only the model's own host dependencies are listed.
dependencies = [
"torch>=2.0",
"numpy>=1.24.4,<2",
"pillow>=9",
"einops>=0.6,<1",
"scipy>=1.10,<2",
"huggingface_hub>=0.30,<2",
]
[project.optional-dependencies]
server = ["fastapi", "uvicorn", "pydantic>=2", "uvloop>=0.19,<1", "httptools>=0.6,<1"]
viz = ["matplotlib"]
test = ["pytest", "pyyaml"] # host tests: also install [server] for the server-equality tests
[project.urls]
Model = "https://huggingface.co/changh95/hamer-p150"
Upstream = "https://github.com/geopavlakos/hamer"
[tool.setuptools.packages.find]
where = ["."]
include = ["hamer", "hamer.*"]
exclude = ["hamer.build*", "hamer.kernels*"]
[tool.setuptools.package-data]
hamer = ["kernels/*.cpp", "kernels/*/*.cpp", "kernels/*/*.hpp", "kernels/*/*.h", "native/*.c"]
|