more tweaks to get cpu to work

app.py

@@ -37,13 +37,14 @@ def download_image(url):
     return PIL.Image.open(BytesIO(response.content)).convert("RGB")
 
 model_path = "runwayml/stable-diffusion-inpainting"
+device = "cpu" # "cuda"
 
 pipe = StableDiffusionInpaintPipeline.from_pretrained(
     model_path,
-    revision="fp16", 
-    torch_dtype=torch.float16,
+    # revision="fp16",
+    # torch_dtype=torch.float16,
     use_auth_token=True
-)
+).to(device)
 
 img_url = "https://cdn.faire.com/fastly/893b071985d70819da5f0d485f1b1bb97ee4f16a6e14ef1bdd4a086b3588be58.png" # wino
 image = download_image(img_url).resize((512, 512))
@@ -53,7 +54,7 @@ prompt = "crazy portal universe"
 
 guidance_scale=7.5
 num_samples = 3
-generator = torch.Generator(device="cpu").manual_seed(0) # change the seed to get different results
+generator = torch.Generator(device=device).manual_seed(0) # change the seed to get different results
 images = pipe(
     prompt=prompt,
     image=image,