 “main” variant of a checkpoint (stored in float32 precision) into mixed precision format, using torch_dtype=torch.float16.


	Copied
from diffusers import DiffusionPipeline
import torch

pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
Now all model components of the pipeline are stored in half-precision dtype. We can now save the
pipeline under a "fp16" variant as follows:


	Copied
pipe.sa