pyproject.toml

[project]
name = "vil-dlm"
version = "0.1.0"
description = "Local smoke-test environment for ViL-DLM"
requires-python = ">=3.11,<3.12"
dependencies = [
  "torch==2.7.1",
  "torchvision==0.22.1",
  "transformers",
  "huggingface_hub",
  "einops",
  "numpy",
  "pillow",
]

[dependency-groups]
dev = [
  "datasets",
  "accelerate",
  "trackio",
  "bitsandbytes>=0.45.0; platform_system == 'Linux'",
]

[tool.uv]
package = false

[tool.uv.sources]
torch = [
  { index = "pytorch-cu128", marker = "platform_system == 'Linux'" },
]
torchvision = [
  { index = "pytorch-cu128", marker = "platform_system == 'Linux'" },
]

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true