File size: 776 Bytes
3a464db | 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 | # Mirrors fast_dvlm/requirements.txt; PyTorch cu124 wheels avoid cu130 driver requirements. --extra-index-url https://download.pytorch.org/whl/cu124 # Core inference / training stack torch==2.6.0 torchvision==0.21.0 transformers==4.57.1 tokenizers>=0.13.3 accelerate>=1.5.0 peft>=0.10.0 datasets==2.14.6 pyarrow==18.0.0 deepspeed>=0.14.4 evaluate==0.4.0 sentencepiece einops>=0.6.1 qwen-vl-utils==0.0.14 Pillow>=10.0.0 numpy tqdm # lmflow vendored under fast_ddrive/lmflow still uses pkg_resources via versioning util. setuptools>=64,<81 cpm_kernels==1.0.11 wandb # Optional: Waymo official ADE / RFS metrics (separate `autovla` env recommended; # tensorflow + waymo_open_dataset conflict with the main inference stack). # tensorflow==2.12.0 # waymo-open-dataset-tf-2-12-0 |