Skip to content

Commit

Permalink
change readme and setup for 0.1.0
Browse files Browse the repository at this point in the history
  • Loading branch information
0xvincii committed Feb 14, 2025
1 parent 60fe37b commit b97379a
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
7 changes: 5 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -72,6 +72,9 @@
## Setup

```bash
conda create -n r1-v python=3.11
conda activate r1-v

bash setup.sh
```

Expand Down Expand Up @@ -99,7 +102,7 @@ bash setup.sh
### GRPO

```bash
cd src/open-r1-multimodal
cd src/r1-v

export DEBUG_MODE="true" # Enable Debug if you want to see the rollout of model during RL
export LOG_PATH="./debug_log_2b.txt"
Expand Down Expand Up @@ -143,7 +146,7 @@ torchrun --nproc_per_node="8" \
We also provide SFT code, please follow the script and edit the config to customize the sft task.

```bash
accelerate launch --config_file src/open-r1-multimodal/configs/zero2.yaml src/open-r1-multimodal/src/open_r1/sft.py --config src/open-r1-multimodal/configs/qwen2vl_sft_config.yaml
accelerate launch --config_file src/r1-v/configs/zero2.yaml src/r1-v/src/open_r1/sft.py --config src/r1-v/configs/qwen2vl_sft_config.yaml
```

## Evaluation
Expand Down
3 changes: 0 additions & 3 deletions setup.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,3 @@
# conda create -n r1-v python=3.11
# conda activate r1-v

# Install the packages in r1-v .
cd src/r1-v
pip install -e ".[dev]"
Expand Down

0 comments on commit b97379a

Please sign in to comment.