[project]
name = "cosmos-robometer-phase21a"
version = "0.1.0"
description = "Reproducible runtime for Phase 2.1A Robometer-Low and Robometer-Low-LoRA"
requires-python = "==3.10.*"
dependencies = [
    "accelerate==1.14.0",
    "bitsandbytes==0.49.2",
    "flash-attn==2.8.3.post1",
    "hf-xet==1.5.1",
    "huggingface-hub==0.36.2",
    "numpy==1.26.4",
    "peft==0.19.1",
    "pillow==11.3.0",
    "pyyaml==6.0.3",
    "qwen-vl-utils==0.0.14",
    "safetensors==0.8.0",
    "torch==2.8.0",
    "torchvision==0.23.0",
    "transformers==4.57.6",
]

[tool.uv]
package = false

# flash-attn must build or select its wheel against the exact PyTorch runtime.
[tool.uv.extra-build-dependencies]
flash-attn = [{ requirement = "torch", match-runtime = true }]

[tool.uv.extra-build-variables]
flash-attn = { FLASH_ATTENTION_SKIP_CUDA_BUILD = "TRUE" }

[tool.uv.sources]
torch = { index = "pytorch-cu128", marker = "sys_platform == 'linux' or sys_platform == 'win32'" }
torchvision = { index = "pytorch-cu128", marker = "sys_platform == 'linux' or sys_platform == 'win32'" }

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