l) —
The device to which the timesteps should be moved to. If None, the timesteps are not moved.    Sets the discrete timesteps used for the diffusion chain (to be run before inference).  step  < source > ( model_output: FloatTensor timestep: int sample: FloatTensor original_image: FloatTensor mask: FloatTensor generator: Optional = None return_dict: bool = True  ) → RePaintSchedulerOutput or tuple  Parameters   model_output (torch.FloatTensor)