no need for xformers on Spaces Zero

app.py

@@ -6,8 +6,10 @@ import numpy as np
 import torch
 from diffusers import LCMScheduler, PixArtAlphaPipeline, Transformer2DModel
 from peft import PeftModel
+import os
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
+IS_SPACE = os.environ.get("SPACE_ID", None) is not None
 
 transformer = Transformer2DModel.from_pretrained(
     "PixArt-alpha/PixArt-XL-2-1024-MS",
@@ -24,7 +26,8 @@ if torch.cuda.is_available():
         transformer=transformer,
         torch_dtype=torch.float16,
     )
-    pipe.enable_xformers_memory_efficient_attention()
+    if not IS_SPACE:
+        pipe.enable_xformers_memory_efficient_attention()
     pipe = pipe.to(device)
 else:
     pipe = PixArtAlphaPipeline.from_pretrained(