Apppublic

AstroAUmin/self-forcing

sourceHugging Faceupdated 22d agoView on Hugging Face

0likes

__init__.py12 linesDownload Raw Back to trainer

1from .diffusion import Trainer as DiffusionTrainer2from .gan import Trainer as GANTrainer3from .ode import Trainer as ODETrainer4from .distillation import Trainer as ScoreDistillationTrainer5 6__all__ = [7    "DiffusionTrainer",8    "GANTrainer",9    "ODETrainer",10    "ScoreDistillationTrainer"11]12