mirror of
https://github.com/leigest519/ScreenCoder.git
synced 2026-02-13 18:22:50 +00:00
Add post-training folder
This commit is contained in:
21
post-training/VLM-R1/setup.sh
Executable file
21
post-training/VLM-R1/setup.sh
Executable file
@@ -0,0 +1,21 @@
|
||||
# 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
|
||||
pip install babel
|
||||
pip install python-Levenshtein
|
||||
pip install matplotlib
|
||||
pip install pycocotools
|
||||
pip install openai
|
||||
pip install httpx[socks]
|
||||
pip install lap scikit-image open-clip-torch playwright
|
||||
playwright install
|
||||
playwright install-deps
|
||||
Reference in New Issue
Block a user