db" \
  --push_to_hub \
  --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
import torch

prior_pipeline = DiffusionPipeline.from_pretrained(output_dir, torch_dtype=torch.float16)
prior_components = {"prior_" + k: v for k,v in prior_pipeline.components.items()}
pipeline =