--report_to="wandb" \
--push_to_hub \
--output_dir="kandi2-decoder-naruto-model"
```  
</hfoption>
</hfoptions>  
Once training is finished, you can use your newly trained model for inference!  
<hfoptions id="training-inference">
<hfoption id="prior model">  
```py
from diffusers import AutoPipelineForText2Image, DiffusionPipeline
import torch