Skip to content

Commit

Permalink
Clean ppo main script (#8596)
Browse files Browse the repository at this point in the history
  • Loading branch information
guoshengCS committed Jun 13, 2024
1 parent 77a7693 commit 4bca376
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 27 deletions.
20 changes: 0 additions & 20 deletions examples/RLHF/ppo_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,26 +16,6 @@
import os
import sys
import types

# os.environ["http_proxy"] = "http://10.162.37.16:8128"
# os.environ["https_proxy"] = "http://10.162.37.16:8128"
# os.environ["no_proxy"] = "localhost,bcebos.com"
# launch would unset http_proxy
# export https_proxy=http://172.19.57.45:3128

# os.environ["http_proxy"] = "http://172.19.56.199:3128"
# os.environ["https_proxy"] = "http://172.19.56.199:3128"

# os.environ["http_proxy"] = "http://172.19.57.45:3128"
# os.environ["https_proxy"] = "http://172.19.57.45:3128"

os.environ["http_proxy"] = "http://10.162.37.16:8128"
os.environ["https_proxy"] = "http://10.162.37.16:8128"
os.environ["no_proxy"] = "localhost,bcebos.com"

# os.environ["http_proxy"] = "agent.baidu.com:8118"
# os.environ["https_proxy"] = "agent.baidu.com:8118"

from dataclasses import dataclass, field
from functools import partial
from typing import Any, Dict, Tuple
Expand Down
7 changes: 0 additions & 7 deletions examples/RLHF/reward_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,13 +32,6 @@
from paddlenlp.transformers import AutoConfig, AutoTokenizer, LlamaTokenizer
from paddlenlp.utils.log import logger

# launch would unset http_proxy
# export https_proxy=http://172.19.57.45:3128
# os.environ["http_proxy"] = "http://172.19.57.45:3128"
# os.environ["https_proxy"] = "http://172.19.57.45:3128"
os.environ["http_proxy"] = "http://10.162.37.16:8128"
os.environ["https_proxy"] = "http://10.162.37.16:8128"


@dataclass
class TrainingArguments(TrainingArguments):
Expand Down

0 comments on commit 4bca376

Please sign in to comment.