github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update init

pull/6119/head
Tong Li 2024-11-14 08:56:24 +00:00
parent 5719974a1e
commit f2f5ff5e24
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
applications/ColossalChat/coati/trainer/__init__.py
View File

@ -5,6 +5,7 @@ from .orpo import ORPOTrainer
from .ppo import PPOTrainer from .ppo import PPOTrainer
from .rm import RewardModelTrainer from .rm import RewardModelTrainer
from .sft import SFTTrainer from .sft import SFTTrainer
from .prm import ProcessRewardModelTrainer
__all__ = [ __all__ = [
"SLTrainer", "SLTrainer",
@ -15,4 +16,5 @@ __all__ = [
"DPOTrainer", "DPOTrainer",
"ORPOTrainer", "ORPOTrainer",
"KTOTrainer", "KTOTrainer",
"ProcessRewardModelTrainer"
] ]