Update app.py

app.py

@@ -29,8 +29,9 @@ def load_pipeline(model_type: str) -> DiffusionPipeline:
     pipe = pipeline_cls.from_pretrained(
         pretrained_model, scheduler=scheduler, torch_dtype=torch.float16, variant="fp16"
     ).to(device)
-    pipe.enable_xformers_memory_efficient_attention()
     pipe.enable_vae_tiling()
+    if torch.cuda.is_available():
+        pipe.enable_xformers_memory_efficient_attention()
     return pipe