v1.1 (Release Candidate 1)
Pre-release
Pre-release
API Updates:
- Add SDE configurations in
configs
package - Introducing
build
function to build non-timed wrapped UNet - Introducing
managers.Manager
, one manager for all separatednn.DiffusionModule
- Introducing
nn.diffusion.FastSamplingDiffusionModule
- Introducing
nn.LatentMode
fornn.LatentDiffusionModule
toencode
,decode
, orforward
- Introducing
sde.SDEType
to load SDEs in default settings. - Introducing the separated
nn.DDPM
andnn.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 ann.TimedModule
, but a normaltorch.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
Release candidate updates:
- Add forward diffusion control in
train_step
andtest_step
function