[project]
name = "nemotron-parse"
version = "1.2.0"
description = "NVIDIA Nemotron-Parse document parsing model"
requires-python = ">=3.10"
dependencies = [
    "transformers>=4.51.3",
    "accelerate==1.12.0",
    "albumentations==2.0.8",
    "timm==1.0.22",
    "einops",
    "Pillow",
    "numpy",
    "opencv-python-headless",
    "beautifulsoup4",
    "open-clip-torch>=3.3.0",
    "pytest>=9.0.3",
]

[project.optional-dependencies]
# 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"]