 > ( model_output: FloatTensor sigma_hat: float sigma_prev: float sample_hat: FloatTensor return_dict: bool = True  ) → ~schedulers.scheduling_karras_ve.KarrasVESchedulerOutput or tuple  Parameters   model_output (torch.FloatTensor) —
The direct output from learned diffusion model.   sigma_hat (float) —   sigma_prev (float) —   sample_hat (torch.FloatTensor) —   return_dict (bool, optional, defaults to True) —
Whether or not to return a ~schedu