is model inherits from SchedulerMixin and ConfigMixin. Check the superclass documentation for the generic
methods the library implements for all schedulers such as loading and saving.  scale_model_input  < source > ( sample: FloatTensor timestep: Optional = None  ) → torch.FloatTensor  Parameters   sample (torch.FloatTensor) —
The input sample.   timestep (int, optional) —
The current timestep in the diffusion chain.   Returns 
torch.FloatTenso