File tree Expand file tree Collapse file tree 5 files changed +1
-79
lines changed
Expand file tree Collapse file tree 5 files changed +1
-79
lines changed Original file line number Diff line number Diff line change @@ -67,7 +67,7 @@ trainer.train()
6767
6868``` python
6969from post_training_toolkit import DiagnosticsCallback
70- from trl import ORPOTrainer
70+ from trl.experimental.orpo import ORPOTrainer
7171
7272trainer = ORPOTrainer(
7373 model = model,
Original file line number Diff line number Diff line change 6767 "NashMDTrainer" , # deprecated import
6868 "OnlineDPOConfig" , # deprecated import
6969 "OnlineDPOTrainer" , # deprecated import
70- "ORPOConfig" , # deprecated import
71- "ORPOTrainer" , # deprecated import
7270 "PPOConfig" , # deprecated import
7371 "PPOTrainer" , # deprecated import
7472 "PRMConfig" , # deprecated import
133131 NashMDTrainer , # deprecated import
134132 OnlineDPOConfig , # deprecated import
135133 OnlineDPOTrainer , # deprecated import
136- ORPOConfig , # deprecated import
137- ORPOTrainer , # deprecated import
138134 PPOConfig , # deprecated import
139135 PPOTrainer , # deprecated import
140136 PRMConfig , # deprecated import
Original file line number Diff line number Diff line change 4343 "nash_md_trainer" : ["NashMDTrainer" ], # deprecated import
4444 "online_dpo_config" : ["OnlineDPOConfig" ], # deprecated import
4545 "online_dpo_trainer" : ["OnlineDPOTrainer" ], # deprecated import
46- "orpo_config" : ["ORPOConfig" ], # deprecated import
47- "orpo_trainer" : ["ORPOTrainer" ], # deprecated import
4846 "ppo_config" : ["PPOConfig" ], # deprecated import
4947 "ppo_trainer" : ["PPOTrainer" ], # deprecated import
5048 "prm_config" : ["PRMConfig" ], # deprecated import
9492 from .nash_md_trainer import NashMDTrainer # deprecated import
9593 from .online_dpo_config import OnlineDPOConfig # deprecated import
9694 from .online_dpo_trainer import OnlineDPOTrainer # deprecated import
97- from .orpo_config import ORPOConfig # deprecated import
98- from .orpo_trainer import ORPOTrainer # deprecated import
9995 from .ppo_config import PPOConfig # deprecated import
10096 from .ppo_trainer import PPOTrainer # deprecated import
10197 from .prm_config import PRMConfig # deprecated import
Load Diff This file was deleted.
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments