cess.
return_dict (bool): option for returning tuple rather than SchedulerOutput class


Returns

SchedulerOutput or tuple



SchedulerOutput if return_dict is True, otherwise a tuple. When
returning a tuple, the first element is the sample tensor.
