trainer/__init__.py

from .diffusion import Trainer as DiffusionTrainer
from .gan import Trainer as GANTrainer
from .ode import Trainer as ODETrainer
from .distillation import Trainer as ScoreDistillationTrainer
from .rewarded_distillation import Trainer as RewardedDistillationTrainer

__all__ = [
    "DiffusionTrainer",
    "GANTrainer",
    "ODETrainer",
    "ScoreDistillationTrainer",
    "RewardedDistillationTrainer"
]