__init__.py 190 Bytes
Newer Older
Fazzie-Maqianli's avatar
Fazzie-Maqianli committed
1
2
3
4
5
6
from .base import Trainer
from .ppo import PPOTrainer
from .rm import RewardModelTrainer
from .sft import SFTTrainer

__all__ = ['Trainer', 'PPOTrainer', 'RewardModelTrainer', 'SFTTrainer']