Added CPU path

app.py

@@ -17,16 +17,16 @@ if deviceStr == "cuda":
                                                               revision="fp16",
                                                               torch_dtype=torch.float16,
                                                               safety_checker=lambda images, **kwargs: (images, False))
+    pipeline.to(device)
+    pipeline.enable_xformers_memory_efficient_attention()
 else:
     pipeline = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting",
-                                                              revision="fp16",
-                                                              torch_dtype=torch.float16,
+                                                              #revision="fp16",
+                                                              #torch_dtype=torch.float16,
                                                               safety_checker=lambda images, **kwargs: (images, False))
 
 #superresolutionPipe = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
 
-pipeline.to(device)
-pipeline.enable_xformers_memory_efficient_attention()
 #pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
 #generator = torch.Generator(device).manual_seed(seed)
 latents = torch.randn((1, 4, 64, 64), device=device)