nsures interchangeability with schedulers that need to scale the denoising model input depending on the
current timestep.  step  < source > ( model_output: FloatTensor timestep: Union sample: FloatTensor generator: Optional = None return_dict: bool = True  ) → ~schedulers.scheduling_consistency_models.ConsistencyDecoderSchedulerOutput or tuple  Parameters   model_output (torch.FloatTensor) —
The direct output from the learned diffusion model.  