ence: PyTorch Flax    Copied from diffusers import StableDiffusionPipeline
import torch

pipeline = StableDiffusionPipeline.from_pretrained("path/to/saved_model", torch_dtype=torch.float16, use_safetensors=True).to("cuda")

image = pipeline(prompt="yoda").images[0]
image.save("yoda-pokemon.png")   Next steps Congratulations on training your own text-to-image model! To learn more about how to use your new model, the following guides may be helpf