Skip to content

GitLab

Unverified Commit 124e76dd authored by realliujiaxu's avatar realliujiaxu Committed by GitHub
Browse files

[docs] add variant="fp16" flag (#4678)

parent 05b0ec63
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
docs/source/en/using-diffusers/inpaint.md
View file @ 124e76dd
...@@ -30,6 +30,7 @@ pipeline = StableDiffusionInpaintPipeline.from_pretrained( ...@@ -30,6 +30,7 @@ pipeline = StableDiffusionInpaintPipeline.from_pretrained(
"runwayml/stable-diffusion-inpainting", "runwayml/stable-diffusion-inpainting",
torch_dtype=torch.float16, torch_dtype=torch.float16,
use_safetensors=True, use_safetensors=True,
variant="fp16",
) )
pipeline = pipeline.to("cuda") pipeline = pipeline.to("cuda")
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment