```  
</hfoption>
</hfoptions>  
Once training is complete, you can use your newly trained model for inference!  
<Tip>  
Can't wait to try your model for inference before training is complete? 🤭 Make sure you have the latest version of 🤗 Accelerate installed.  
```py
from diffusers import DiffusionPipeline, UNet2DConditionModel
from transformers import CLIPTextModel
import torch