rary implements for all schedulers such as loading and saving.  convert_model_output  < source > ( model_output: FloatTensor *args sample: FloatTensor = None **kwargs  ) → torch.FloatTensor  Parameters   model_output (torch.FloatTensor) —
The direct output from the learned diffusion model.   timestep (int) —
The current discrete timestep in the diffusion chain.   sample (torch.FloatTensor) —
A current instance of a sample created by the diffusi