@spaces.GPU(duration=400)

app.py

@@ -141,7 +141,8 @@ def generate_image_on_gpu(args):
         print(f"Error with {model_name} on GPU {gpu_id}: {str(e)}")
         release_gpu(gpu_id)
         raise e
-
+        
+@spaces.GPU(duration=400)
 def generate_all(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress()):
     outputs = [None] * (len(MODEL_CONFIGS) * 2)