ep (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_timesteps

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

)


Parameters 

num_inference_steps (Optional[int]) —
