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 .rm import RewardModelTrainer
from .sft import SFTTrainer
from .prm import ProcessRewardModelTrainer
__all__ = [
"SLTrainer",
@ -15,4 +16,5 @@ __all__ = [
"DPOTrainer",
"ORPOTrainer",
"KTOTrainer",
"ProcessRewardModelTrainer"
]