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

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
pull/6119/head
pre-commit-ci[bot] 2024-11-14 08:59:30 +00:00
parent 852333423d
commit ab992b89e4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3,9 +3,9 @@ from .dpo import DPOTrainer
from .kto import KTOTrainer from .kto import KTOTrainer
from .orpo import ORPOTrainer from .orpo import ORPOTrainer
from .ppo import PPOTrainer from .ppo import PPOTrainer
from .prm import ProcessRewardModelTrainer
from .rm import RewardModelTrainer from .rm import RewardModelTrainer
from .sft import SFTTrainer from .sft import SFTTrainer
from .prm import ProcessRewardModelTrainer
__all__ = [ __all__ = [
"SLTrainer", "SLTrainer",
@ -16,5 +16,5 @@ __all__ = [
"DPOTrainer", "DPOTrainer",
"ORPOTrainer", "ORPOTrainer",
"KTOTrainer", "KTOTrainer",
"ProcessRewardModelTrainer" "ProcessRewardModelTrainer",
] ]