R1-Onevision/VLM-R1/setup.sh
2025-04-13 07:27:40 +00:00

13 lines
373 B
Bash
Executable File

# conda create -n vlm-r1 python=3.11
# conda activate vlm-r1
# Install the packages in open-r1-multimodal .
cd src/open-r1-multimodal # We edit the grpo.py and grpo_trainer.py in open-r1 repo.
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