Skip to content

Commit

Permalink
clean readme.md
Browse files Browse the repository at this point in the history
  • Loading branch information
hijkzzz committed Feb 18, 2024
1 parent 50c79b4 commit c985657
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,6 @@ cd /openrlhf/examples/scripts

# build OpenRLHF (i.e, pip install)
./build_openrlhf.sh

# due to the compatibility of nVIDIA PyTorch image
pip uninstall xgboost transformer_engine -y

Expand All @@ -169,17 +168,15 @@ pip uninstall xgboost transformer_engine -y

# launch the master node of ray in container
ray start --head --node-ip-address 0.0.0.0 --num-gpus 8

# if you want to launch ray on more nodes, use
ray start --address {MASTER-NODE-ADDRESS}:6379 --num-gpus 8


# train ray PPO model, requires 8 gpus in default config
./train_ppo_llama_ray.sh

# for 70B models and vLLM-based RLHF
# for 70B models and vLLM-based RLHF (important!)
pip install vllm==0.2.4

# due to the compatibility of vLLM
pip uninstall flash_attn -y

Expand Down

0 comments on commit c985657

Please sign in to comment.