pipe.enable_model_cpu_offload()
prompt="A robot naruto, 4k photo"
image = pipeline(prompt=prompt, negative_prompt=negative_prompt).images[0]
```  
<Tip>  
Feel free to replace `kandinsky-community/kandinsky-2-2-decoder` with your own trained decoder checkpoint!  
</Tip>  
</hfoption>
<hfoption id="decoder model">  
```py
from diffusers import AutoPipelineForText2Image
import torch