anton-l HF staff commited on
Commit
0bbf937
1 Parent(s): caac104

Add `.to("cuda")` to the example snippet for float16 compatibility

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -42,6 +42,7 @@ pipe = StableDiffusionInpaintPipeline.from_pretrained(
42
  revision="fp16",
43
  torch_dtype=torch.float16,
44
  )
 
45
  prompt = "Face of a yellow cat, high resolution, sitting on a park bench"
46
  #image and mask_image should be PIL images.
47
  #The mask structure is white for inpainting and black for keeping as is
42
  revision="fp16",
43
  torch_dtype=torch.float16,
44
  )
45
+ pipe.to("cuda")
46
  prompt = "Face of a yellow cat, high resolution, sitting on a park bench"
47
  #image and mask_image should be PIL images.
48
  #The mask structure is white for inpainting and black for keeping as is