_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.
For more details, see the original paper: https://arxiv.org/abs/2202.09778

scale_model_input

<
source
>
(
sample: FloatTensor
*args
**kwargs

)
→
torch.FloatTensor

Parameters 

sample (torch.FloatTensor) — input sample


Returns

t