p or scaled input timestep array.
  Gets scaled timesteps from the Karras sigmas for input to the consistency model.  step  < source > ( model_output: FloatTensor timestep: Union sample: FloatTensor generator: Optional = None return_dict: bool = True  ) → CMStochasticIterativeSchedulerOutput or tuple  Parameters   model_output (torch.FloatTensor) —
The direct output from the learned diffusion model.   timestep (float) —
The current timestep in 