multigpu longer timeout

This commit is contained in:
Wing Lian
2025-04-09 01:54:35 -04:00
parent 9f69597a5f
commit 37a66e6866

View File

@@ -68,7 +68,7 @@ def run_cmd(cmd: str, run_folder: str):
@app.function(
image=cicd_image,
gpu=GPU_CONFIG,
timeout=60 * 60,
timeout=90 * 60,
cpu=8.0,
memory=131072 * N_GPUS,
volumes=VOLUME_CONFIG,