From 124e76ddc62778b5ebfd3bb387e4578e5d8d3237 Mon Sep 17 00:00:00 2001 From: realliujiaxu <72550220+realliujiaxu@users.noreply.github.com> Date: Thu, 24 Aug 2023 01:00:34 +0800 Subject: [PATCH] [docs] add variant="fp16" flag (#4678) --- docs/source/en/using-diffusers/inpaint.md | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/source/en/using-diffusers/inpaint.md b/docs/source/en/using-diffusers/inpaint.md index d5ae9e2665..7df755e322 100644 --- a/docs/source/en/using-diffusers/inpaint.md +++ b/docs/source/en/using-diffusers/inpaint.md @@ -30,6 +30,7 @@ pipeline = StableDiffusionInpaintPipeline.from_pretrained( "runwayml/stable-diffusion-inpainting", torch_dtype=torch.float16, use_safetensors=True, + variant="fp16", ) pipeline = pipeline.to("cuda") ```