d via scheduler.config.num_train_timesteps.
SchedulerMixin provides general loading and saving functionality via the SchedulerMixin.save_pretrained() and
from_pretrained() functions.

get_lms_coefficient

<
source
>
(
order
t
current_order

)


Parameters 

order (TODO) —


t (TODO) —


current_order (TODO) —



Compute a linear multistep coefficient.

scale_model_input

<
source
>
(
sample: FloatTensor
timestep: typing.Union[float, torch.Float