AutomatosX's picture
Pin renamed CUDA TF32 inference source and preserve strict artifact verification
7fb4118 verified
Raw History Blame Contribute Delete
964 Bytes
[build-system]
requires = ["setuptools>=77", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "ax-doclayout"
version = "1.2.0.post3"
description = "AX DocLayout page-raster layout inference."
readme = "README.md"
requires-python = ">=3.12,<3.14"
license = "AGPL-3.0-only"
license-files = ["LICENSE", "NOTICE"]
authors = [{ name = "AutomatosX" }]
dependencies = ["huggingface-hub>=0.36,<2", "numpy>=1.26,<3", "pillow>=12.2,<13"]
[project.optional-dependencies]
onnx = [
"onnxruntime-gpu[cuda,cudnn]==1.23.2",
"nvidia-cuda-runtime-cu12==12.8.90",
"nvidia-cuda-nvrtc-cu12==12.8.93",
"nvidia-cufft-cu12==11.3.3.83",
"nvidia-curand-cu12==10.3.9.90",
"nvidia-cublas-cu12==12.8.4.1",
"nvidia-cudnn-cu12==9.10.2.21",
]
torch = ["torch>=2.12,<2.13", "safetensors>=0.4,<1"]
[project.scripts]
ax-doclayout = "ax_doclayout.cli:main"
[tool.setuptools.packages.find]
where = ["src"]
[tool.setuptools.package-data]
ax_doclayout = ["py.typed"]