1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

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

This commit is contained in:
realliujiaxu
2023-08-24 01:00:34 +08:00
committed by GitHub
parent 05b0ec63bc
commit 124e76ddc6

View File

@@ -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")
```