Tensor

Parameters 

Ensures interchangeability with schedulers that need to scale the denoising model input depending on the —


current timestep. —
sample (torch.FloatTensor): input sample timestep (int, optional): current timestep


Returns

torch.FloatTensor



scaled input sample



set_timesteps

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

)


Param