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) —