[project]
name = "nemotron-parse"
version = "2.0.0"
description = "NVIDIA Nemotron-Parse document parsing model"
requires-python = ">=3.10"
dependencies = [
    "transformers==5.6.1",
    "accelerate==1.12.0",
    "timm==1.0.22",
    "einops",
    "Pillow",
    "numpy",
    "opencv-python-headless",
    "beautifulsoup4",
    "pytest>=9.0.3",
]

[project.optional-dependencies]
# Direct Transformers inference currently loads C-RADIO remote code, whose
# optional OpenCLIP adaptor is inspected during module validation.
transformers-inference = ["open-clip-torch==3.2.0"]
# vLLM serving (install separately in the serving container).
vllm = ["openai"]
# Development / testing.
dev = ["pytest"]

[tool.uv]
# This repo is a model directory loaded via trust_remote_code, not an
# installable Python package, so uv should only manage dependencies.
package = false

# torch and torchvision ship pre-compiled with CUDA support inside the
# NVIDIA base image (nvcr.io/nvidia/pytorch:*).  uv must not overwrite them.
exclude-dependencies = ["torch", "torchvision"]