hot-fix: update PPOConfig import in examples (#352)

jon-tow · web-flow · commit 291c77432206 · 2023-03-03T14:07:32.000-05:00
diff --git a/examples/summarize_rlhf/trlx_gptj_text_summarization.py b/examples/summarize_rlhf/trlx_gptj_text_summarization.py
@@ -16,7 +16,7 @@
     TrainConfig,
     TRLConfig,
 )
-from trlx.trainer.nn.ppo_models import PPOConfig
+from trlx.models.modeling_ppo import PPOConfig
 
 REWARD_CHECKPOINT_PATH = "reward_model/rm_checkpoint/pytorch_model.bin"
 if not os.path.exists(REWARD_CHECKPOINT_PATH):

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@`
`16`	`16`	`TrainConfig,`
`17`	`17`	`TRLConfig,`
`18`	`18`	`)`
`19`		`-from trlx.trainer.nn.ppo_models import PPOConfig`
	`19`	`+from trlx.models.modeling_ppo import PPOConfig`
`20`	`20`
`21`	`21`	`REWARD_CHECKPOINT_PATH = "reward_model/rm_checkpoint/pytorch_model.bin"`
`22`	`22`	`if not os.path.exists(REWARD_CHECKPOINT_PATH):`