cuda check

app.py

@@ -45,12 +45,13 @@ pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
 
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
-# Remove if you do not have xformers installed
-# see https://huggingface.co/docs/diffusers/v0.13.0/en/optimization/xformers#installing-xformers
-# for installation instructions
-pipe.enable_xformers_memory_efficient_attention()
+if torch.cuda.is_available():
+    # Remove if you do not have xformers installed
+    # see https://huggingface.co/docs/diffusers/v0.13.0/en/optimization/xformers#installing-xformers
+    # for installation instructions
+    pipe.enable_xformers_memory_efficient_attention()
 
-pipe.to('cuda')
+    pipe.to('cuda')
 
 # Functions