azula.plugins.vdm¶
Velocity diffusion model (VDM) plugin.
from azula.plugins import vdm
References
Classes¶
Creates a velocity denoiser. |
Functions¶
Loads a pre-trained VDM denoiser. |
Descriptions¶
- class azula.plugins.vdm.VelocityDenoiser(backbone, schedule=None)¶[source]
Creates a velocity denoiser.
- Parameters:
backbone (Module) – A time conditional network.
schedule (Schedule) – A noise schedule. If
None, useazula.noise.VPScheduleinstead.
- azula.plugins.vdm.load_model(name, **kwargs)¶[source]
Loads a pre-trained VDM denoiser.
- Parameters:
name (str) – The pre-trained model name.
kwargs – Keyword arguments passed to
torch.load.
- Returns:
A pre-trained denoiser.
- Return type: