From 1647b06ebea4eb1dfbf8076b8700a2406464de2c Mon Sep 17 00:00:00 2001
From: Charles Frye <charles@modal.com>
Date: Thu, 3 Oct 2024 14:14:07 -0400
Subject: [PATCH] adjust timeout for compilation of flux (#913)

---
 06_gpu_and_ml/stable_diffusion/flux.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/06_gpu_and_ml/stable_diffusion/flux.py b/06_gpu_and_ml/stable_diffusion/flux.py
index 4af46bedb..1fdaac48f 100644
--- a/06_gpu_and_ml/stable_diffusion/flux.py
+++ b/06_gpu_and_ml/stable_diffusion/flux.py
@@ -111,6 +111,7 @@
 @app.cls(
     gpu="H100",  # FA3 is tuned for Hopper
     container_idle_timeout=20 * MINUTES,
+    timeout=60 * MINUTES,  # leave plenty of time for compilation
     volumes={  # add Volumes to store serializable compilation artifacts
         "/root/.nv": modal.Volume.from_name("nv-cache", create_if_missing=True),
         "/root/.triton": modal.Volume.from_name(