methods the library implements for all schedulers such as loading and saving.  scale_model_input  < source > ( sample: FloatTensor timestep: Union  ) → torch.FloatTensor  Parameters   sample (torch.FloatTensor) —