v1.1
API Updates:
- Add forward diffusion control in
train_stepandtest_stepfunction - Add SDE configurations in
configspackage - Introducing
buildfunction to build non-timed wrapped UNet - Introducing
managers.Manager, one manager for all separatednn.DiffusionModule - Introducing
nn.diffusion.FastSamplingDiffusionModule - Introducing
nn.LatentModefornn.LatentDiffusionModuletoencode,decode, orforward - Introducing
sde.SDETypeto load SDEs in default settings. - Introducing the separated
nn.DDPMandnn.LatentDiffusionModule - Introducing the separated
nn.DiffusionModulefor separated diffusion process into PyTorch module - Introducing the separated
nn.SDEModule - Multi-GPUs support for
nn.DiffusionModule - Parsing additional arguments when forwarding in
nn.diffusion.LatentDiffusionModule - The new separated
nn.DiffusionModuleno longer need ann.TimedModule, but a normaltorch.nn.Moduleinstead for better compatibility on other unets - Use
prior_samplingin SDE instead of direct randomize when adding noises.
Other updates:
- Compatibility for non-wrapped models improved
- Minor bugs fixed
- Typing improvement