File size: 397 Bytes
22e5669 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# Install the packages in r1-v .
cd src/r1-v
pip install -e ".[dev]"
# Addtional modules
pip install wandb==0.18.3
pip install tensorboardx
pip install qwen_vl_utils torchvision
pip install flash-attn --no-build-isolation
# vLLM support
pip install vllm==0.7.2
# fix transformers version
pip install git+https://github.com/huggingface/transformers.git@336dc69d63d56f232a183a3e7f52790429b871ef |