Update app.py

app.py

@@ -61,7 +61,7 @@ def get_scheduler(scheduler_config: Dict, name: str) -> Optional[Callable]:
     }
     return scheduler_factory_map.get(name, lambda: None)()
 
-@spaces.GPU
+@spaces.GPU(enable_queue=False)
 def generate(
     prompt: str,
     negative_prompt: str = None,
@@ -69,7 +69,7 @@ def generate(
     width: int = 1024,
     height: int = 1024,
     guidance_scale: float = 5.0,
-    num_inference_steps: int = 24,
+    num_inference_steps: int = 26,
     sampler: str = "Euler a",
     clip_skip: int = 1,
     progress=gr.Progress(track_tqdm=True),
@@ -84,7 +84,6 @@ def generate(
             add_watermarker=False,
             use_auth_token=HF_TOKEN
         )
-    pipe.to(device)
     
     generator = seed_everything(seed)
     pipe.scheduler = get_scheduler(pipe.scheduler.config, sampler)
@@ -94,6 +93,8 @@ def generate(
         num_hidden_layers = 12 - (clip_skip - 1),
         torch_dtype = torch.float16
     )
+    
+    pipe.to(device)
 
     try: