Skip to content

v1.1

Latest
Compare
Choose a tag to compare
@kisonho kisonho released this 09 Sep 13:50

API Updates:

  • Add forward diffusion control in train_step and test_step function
  • Add SDE configurations in configs package
  • Introducing build function to build non-timed wrapped UNet
  • Introducing managers.Manager, one manager for all separated nn.DiffusionModule
  • Introducing nn.diffusion.FastSamplingDiffusionModule
  • Introducing nn.LatentMode for nn.LatentDiffusionModule to encode, decode, or forward
  • Introducing sde.SDEType to load SDEs in default settings.
  • Introducing the separated nn.DDPM and nn.LatentDiffusionModule
  • Introducing the separated nn.DiffusionModule for 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.DiffusionModule no longer need a nn.TimedModule, but a normal torch.nn.Module instead for better compatibility on other unets
  • Use prior_sampling in SDE instead of direct randomize when adding noises.

Other updates:

  • Compatibility for non-wrapped models improved
  • Minor bugs fixed
  • Typing improvement