zeroshotGPU / pyproject.toml
Arjunvir Singh
Fix ZeroGPU state-loss in EmbeddingRetriever; add einops + smoke surfaces
52764bf
[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "zero-shot-gpu-doc-parser"
version = "0.1.0"
description = "Zero-shot GPU document parsing and agentic chunking control plane."
readme = "README.md"
requires-python = ">=3.11"
license = { text = "MIT" }
authors = [{ name = "Zero-Shot GPU Document Parser Contributors" }]
dependencies = []
[project.optional-dependencies]
pdf = ["pymupdf>=1.24.0,<1.28.0"]
yaml = ["pyyaml>=6.0.1,<7.0.0"]
docling = ["docling>=2.0.0,<3.0.0"]
# `spaces` mirrors requirements.txt at the root, which is what HF Spaces
# installs verbatim. Keep these two in sync; torch is intentionally absent
# because the l4x1 Space image preinstalls a CUDA-matched build.
spaces = [
"gradio>=4.44.0,<7.0.0",
"pymupdf>=1.24.0,<1.28.0",
"pyyaml>=6.0.1,<7.0.0",
"docling>=2.0.0,<3.0.0",
]
embedding = [
"sentence-transformers>=3.0.0,<4.0.0",
"transformers>=4.45.0,<6.0.0",
# jinaai/jina-embeddings-v3's custom modeling needs einops; not pulled
# in transitively by sentence-transformers.
"einops>=0.7.0",
]
gpu_repair = ["transformers>=4.45.0,<6.0.0"]
dev = ["pytest>=8.0.0"]
[project.scripts]
zsgdp = "zsgdp.cli:main"
[tool.setuptools.packages.find]
where = ["."]
include = ["zsgdp*"]
[tool.pytest.ini_options]
testpaths = ["tests"]
pythonpath = ["."]