anton-l HF staff commited on
Commit
349f095
1 Parent(s): 781cb3e

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
@@ -48,6 +48,7 @@ pipe = StableDiffusionInpaintPipeline.from_pretrained(
48
  "stabilityai/stable-diffusion-2-inpainting",
49
  torch_dtype=torch.float16,
50
  )
 
51
  prompt = "Face of a yellow cat, high resolution, sitting on a park bench"
52
  #image and mask_image should be PIL images.
53
  #The mask structure is white for inpainting and black for keeping as is
 
48
  "stabilityai/stable-diffusion-2-inpainting",
49
  torch_dtype=torch.float16,
50
  )
51
+ pipe.to("cuda")
52
  prompt = "Face of a yellow cat, high resolution, sitting on a park bench"
53
  #image and mask_image should be PIL images.
54
  #The mask structure is white for inpainting and black for keeping as is