nction, such as num_train_timesteps. They can be accessed via scheduler.config.num_train_timesteps.
SchedulerMixin provides general loading and saving functionality via the SchedulerMixin.save_pretrained() and
from_pretrained() functions.

convert_model_output

<
source
>
(
model_output: FloatTensor
timestep: int
sample: FloatTensor

)
→
torch.FloatTensor

Parameters 

model_output (torch.FloatTensor) — direct output from learned diffusion mode