andb" \
  --push_to_hub \
  --output_dir="wuerstchen-prior-pokemon-model" Once training is complete, you can use your newly trained model for inference!   Copied import torch
from diffusers import AutoPipelineForText2Image
from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS

pipeline = AutoPipelineForText2Image.from_pretrained("path/to/saved/model", torch_dtype=torch.float16).to("cuda")

caption = "A cute bird pokemon holding a