File size: 720 Bytes
2b05eb6 75071e1 2b05eb6 0d77b0a 2b05eb6 75071e1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | [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
|