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 .naive_cd import Trainer as ConsistencyDistillationTrainer

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