| 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" | |
| ] | |