 according to a gamma_i ≥ 0 to reach a
higher noise level sigma_hat = sigma_i + gamma_i*sigma_i.  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 interchangeability with schedulers that need to s