g and saving functionality via the SchedulerMixin.save_pretrained() and
from_pretrained() functions.

scale_model_input

<
source
>
(
sample: FloatTensor
timestep: typing.Union[float, torch.FloatTensor]

)
→
torch.FloatTensor

Parameters 

sample (torch.FloatTensor) — input sample


timestep (float or torch.FloatTensor) — the current timestep in the diffusion chain


Returns

torch.FloatTensor



scaled input sample


Scales the denoising model