loatTensor

Parameters 

sample (torch.FloatTensor) — input sample


timestep (int, optional) — current timestep


Returns

torch.FloatTensor



scaled input sample


Ensures interchangeability with schedulers that need to scale the denoising model input depending on the
current timestep.

set_sigmas

<
source
>
(
num_inference_steps: int
sigma_min: float = None
sigma_max: float = None
sampling_eps: float = None

)


Parameters 

num_inference_