| [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"] | |