latents = latents * scheduler.init_noise_sigma