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

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

parent 05b0ec63
......@@ -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")
```
......
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