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

README.md

@@ -42,6 +42,7 @@ pipe = StableDiffusionInpaintPipeline.from_pretrained(
     revision="fp16",
     torch_dtype=torch.float16,
 )
+pipe.to("cuda")
 prompt = "Face of a yellow cat, high resolution, sitting on a park bench"
 #image and mask_image should be PIL images.
 #The mask structure is white for inpainting and black for keeping as is