mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
382 B
19 lines
382 B
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",
|
|
]
|