lerMixin provides general loading and saving
functionality via the SchedulerMixin.save_pretrained() and from_pretrained() functions.  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.FloatTensor
  
A scaled input sample.
  Ensures interchangeab