  --output_dir="kandi2-prior-pokemon-model"   Once training is finished, you can use your newly trained model for inference! prior model decoder model    Copied from diffusers import AutoPipelineForText2Image, DiffusionPipeline