>>> scheduler = DDPMScheduler.from_pretrained("google/ddpm-cat-256")
>>> model = UNet2DModel.from_pretrained("google/ddpm-cat-256", use_safetensors=True).to("cuda")
```  
2. Set the number of timesteps to run the denoising process for:  
```py
>>> scheduler.set_timesteps(50)
```