image.save("naruto.png")
```  
</hfoption>
<hfoption id="PyTorch XLA">  
[PyTorch XLA](https://pytorch.org/xla) allows you to run PyTorch on XLA devices such as TPUs, which can be faster. The initial warmup step takes longer because the model needs to be compiled and optimized. However, subsequent calls to the pipeline on an input **with the same length** as the original prompt are much faster because it can reuse the optimized graph.  
```py
from diffusers import DiffusionPipeline
import torch