pyproject.toml

[build-system]
requires = ["setuptools>=69", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "parvl"
version = "0.1.0"
description = "Parallel scaling and expandable compute allocation for multimodal LLMs"
readme = "README.md"
requires-python = ">=3.10"
license = {text = "MIT"}
dependencies = [
  "torch>=2.6",
  "torchvision>=0.21",
  "transformers==4.55.0",
  "safetensors>=0.4,<1",
  "timm>=1.0.19,<2",
  "einops>=0.8,<1",
  "numpy>=1.26,<3",
  "Pillow>=10,<12",
  "packaging>=24,<26",
]

[project.optional-dependencies]
train = [
  "accelerate>=1.7,<2",
  "deepspeed>=0.17.4,<0.19",
  "datasets>=3.6,<5",
  "peft>=0.14,<1",
  "opencv-python-headless>=4.10,<5",
  "imageio>=2.34,<3",
  "decord==0.6.0",
  "sentencepiece>=0.2,<1",
  "tensorboard>=2.18,<3",
]
storage = [
  "boto3>=1.35,<2",
  "mmengine>=0.10,<1",
]
dev = ["pytest>=8,<10"]

[tool.setuptools.packages.find]
where = ["."]
include = ["internvl*"]
namespaces = true

[tool.pytest.ini_options]
addopts = "-ra"
pythonpath = ["."]
testpaths = ["tests"]