mirror of https://github.com/hpcaitech/ColossalAI
19 lines
382 B
Python
Executable File
19 lines
382 B
Python
Executable File
from .base import OLTrainer, SLTrainer
|
|
from .dpo import DPOTrainer
|
|
from .kto import KTOTrainer
|
|
from .orpo import ORPOTrainer
|
|
from .ppo import PPOTrainer
|
|
from .rm import RewardModelTrainer
|
|
from .sft import SFTTrainer
|
|
|
|
__all__ = [
|
|
"SLTrainer",
|
|
"OLTrainer",
|
|
"RewardModelTrainer",
|
|
"SFTTrainer",
|
|
"PPOTrainer",
|
|
"DPOTrainer",
|
|
"ORPOTrainer",
|
|
"KTOTrainer",
|
|
]
|