Update app.py

app.py

@@ -31,7 +31,7 @@ def single_condition_generate_image(prompt, spatial_img, height, width, seed, co
     # Set the control type
     if control_type == "Ghibli":
         lora_path = os.path.join(lora_base_path, "Ghibli.safetensors")
-    set_single_lora(pipe.transformer, lora_path, lora_weights=[1], cond_size=512)
+    set_single_lora(pipe.transformer, lora_path, lora_weights=[1], cond_size=512, device="cpu")
     
     # Process the image
     spatial_imgs = [spatial_img] if spatial_img else []