loat) —   sample_hat (torch.FloatTensor) —   return_dict (bool, optional, defaults to True) —
Whether or not to return a ~schedulers.scheduling_karras_ve.KarrasVESchedulerOutput or tuple.   Returns 
~schedulers.scheduling_karras_ve.KarrasVESchedulerOutput or tuple
  
If return_dict is True, ~schedulers.scheduling_karras_ve.KarrasVESchedulerOutput is returned,
otherwise a tuple is returned where the first element is the sample tensor.
  Predict 