pyproject.toml

[project]
name = "visual-retrieval-colpali"
version = "0.1.0"
description = "Visual retrieval with ColPali"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "Apache-2.0" }
dependencies = [
    "python-fasthtml",
    "huggingface-hub",
    "pyvespa@git+https://github.com/vespa-engine/pyvespa",
    "vespacli",
    "torch",
    "vidore-benchmark[interpretability]>=4.0.0,<5.0.0",
    "colpali-engine",
    "einops",
    "pypdf",
    "setuptools",
    "python-dotenv",
    "shad4fast>=1.2.1",
]

# dev-dependencies
[project.optional-dependencies]
dev = [
    "ruff",
    "python-dotenv",
    "huggingface_hub[cli]"
]