timestep
schedule is used. If timesteps is passed, num_inference_steps must be None.    Sets the discrete timesteps used for the diffusion chain (to be run before inference).  step  < source > ( model_output: FloatTensor timestep: int sample: FloatTensor generator: Optional = None return_dict: bool = True  ) → ~schedulers.scheduling_utils.LCMSchedulerOutput or tuple  Parameters   model_output (torch.FloatTensor) —
The direct output from learned