| [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 | |