File size: 276 Bytes
54c5666 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
"""
Training components and utilities
"""
from .distributed_4d import DistributedConfig, DistributedTrainer
from .rlhf_advanced import RLHF2System, RLHFConfig
__all__ = [
"DistributedConfig",
"DistributedTrainer",
"RLHF2System",
"RLHFConfig"
]
|