Skip to main content
This node is designed to enhance a model’s sampling capabilities by integrating continuous EDM (Energy-based Diffusion Models) sampling techniques. It allows for the dynamic adjustment of the noise levels within the model’s sampling process, offering a more refined control over the generation quality and diversity.

Inputs

ParameterData TypePython dtypeDescription
modelMODELtorch.nn.ModuleThe model to be enhanced with continuous EDM sampling capabilities. It serves as the foundation for applying the advanced sampling techniques.
samplingCOMBO[STRING]strSpecifies the type of sampling to be applied, either ‘eps’ for epsilon sampling or ‘v_prediction’ for velocity prediction, influencing the model’s behavior during the sampling process.
sigma_maxFLOATfloatThe maximum sigma value for noise level, allowing for upper bound control in the noise injection process during sampling.
sigma_minFLOATfloatThe minimum sigma value for noise level, setting the lower limit for noise injection, thus affecting the model’s sampling precision.

Outputs

ParameterData TypePython dtypeDescription
modelMODELtorch.nn.ModuleThe enhanced model with integrated continuous EDM sampling capabilities, ready for further use in generation tasks.