 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.

step

<
source
>
(
model_output: FloatTensor
timestep: int
sample: FloatTensor
original_image: FloatTensor
mask: FloatTensor
generator: typing.Optional[torch._C.Generator] = None
return_dict: bool = True

)
→
~schedu