# PyTorch: cu124 wheels (same stack as conda env ``dvlm``: torch 2.6.0+cu124). Avoids cu130
# wheels that require a newer NVIDIA user-space driver than typical 12.2 installs.
--extra-index-url https://download.pytorch.org/whl/cu124

# Inference / chatbot side.
torch==2.6.0
torchvision==0.21.0
qwen-vl-utils==0.0.14
gradio>=3.35.2,<4.0
Pillow>=10.0.0
openai>=1.0.0

# Training (Fast-dVLM finetuning via vendored lmflow under ``third_party/lmflow``).
# Mirrors v2/requirements.txt — the lmflow code paths still rely on
# ``transformers.utils.send_example_telemetry`` etc. that newer transformers removed.
# See fast_dvlm/train_scripts/finetune_multimodal_example.sh.
transformers==4.57.1
tokenizers>=0.13.3
datasets==2.14.6
pyarrow==18.0.0
accelerate>=0.27.2
peft>=0.10.0
deepspeed>=0.14.4
evaluate==0.4.0
sentencepiece
einops>=0.6.1
cpm_kernels==1.0.11
wandb
# lmflow.utils.versioning still imports pkg_resources; pin setuptools accordingly.
setuptools>=64,<81