Skip to content

v0.2.4: Patch release

Compare
Choose a tag to compare
@anton-l anton-l released this 22 Aug 17:09
· 3826 commits to main since this release

This patch release allows the Stable Diffusion pipelines to be loaded with float16 precision:

pipe = StableDiffusionPipeline.from_pretrained(
           "CompVis/stable-diffusion-v1-4", 
           revision="fp16", 
           torch_dtype=torch.float16, 
           use_auth_token=True
)
pipe = pipe.to("cuda")

The resulting models take up less than 6900 MiB of GPU memory.