Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 
 # Load Stable Diffusion XL base model with LoRA weights (requires PEFT backend)
 pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
-pipe.to("cuda")
+pipe.to("cpu")
 
 # Ensure PEFT is installed and compatible
 try:
@@ -14,6 +14,8 @@ try:
 except ImportError:
     raise ImportError("PEFT is required for loading LoRA weights. Install it using `pip install peft`." )
 
+pipe.unet.set_default_attn_processor()
+pipe.vae.set_default_attn_processor()
 
 def infer(color_prompt, dress_type_prompt, design_prompt):
     prompt = (