File size: 464 Bytes
bb7f76d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# 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
pip install nltk
pip install rouge_score
pip install deepspeed
# fix transformers version
# pip install git+https://github.com/huggingface/transformers.git@336dc69d63d56f232a183a3e7f52790429b871ef
|