dicted noise), and calls step_prk()
or step_plms() depending on the internal variable counter.  step_plms  < source > ( model_output: FloatTensor timestep: int sample: FloatTensor return_dict: bool = True  ) → SchedulerOutput or tuple  Parameters   model_output (torch.FloatTensor) —
The direct output from learned diffusion model.   timestep (int) —
The current discrete timestep in the diffusion chain.   sample (torch.FloatTensor) —
A current in