Tensor) — the current timestep in the diffusion chain


Returns

torch.FloatTensor



scaled input sample


Scales the denoising model input by (sigma**2 + 1) ** 0.5 to match the Euler algorithm.

set_timesteps

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

)


Parameters 

num_inference_steps (int) —
the number of diffusion steps used when generating samples with a pre-trained model.


device (str or tor