> ( sample: FloatTensor timestep: Union  ) → torch.FloatTensor  Parameters   sample (torch.FloatTensor) —
The input sample.   timestep (float or torch.FloatTensor) —
The current timestep in the diffusion chain.   Returns 
torch.FloatTensor
  
A scaled input sample.
  Scales the consistency model input by (sigma**2 + sigma_data**2) ** 0.5.  set_timesteps  < source > ( num_inference_steps: Optional = None device: Union = None timesteps: Optional 