adjust timeout for compilation of flux (#913)

modal-labs · Oct 3, 2024 · 1647b06 · 1647b06
1 parent 85665a6
commit 1647b06
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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(