- `nn.DiffusionModule` no longer need a `nn.TimedModule`, but a normal `torch.nn.Module` instead for better compatibility on other unets