scheduler.set_timesteps(num_inference_steps)