le: FloatTensor
*args
**kwargs

)
→
torch.FloatTensor

Parameters 

sample (torch.FloatTensor) — input sample


Returns

torch.FloatTensor



scaled input sample


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

set_timesteps

<
source
>
(
num_inference_steps: int
device: typing.Union[str, torch.device] = None

)


Parameters 

num_inference_steps (int) —
the number of 