maybe cuda

app.py

@@ -87,6 +87,8 @@ pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.conf
 pipe.unet.set_attn_processor(AttnProcessor2_0())
 if device == "cuda":
     pipe.enable_sequential_cpu_offload(gpu_id=0, device="cuda")
+else:
+    pipe = pipe.to("cuda")  # for spaces
 
 
 def get_aspect_ratio(width: int, height: int) -> str:
@@ -140,7 +142,7 @@ def format_prompt(prompt: str, prompt_suffix: str):
     return f"{prompt}, {prompt_suffix}"
 
 
-@spaces.GPU
+@spaces.GPU(duration=30)
 def generate_image(
     prompt: str,
     negative_prompt: str,