diff --git a/src/gpu/modal_export.py b/src/gpu/modal_export.py index 3300053..9e0b933 100644 --- a/src/gpu/modal_export.py +++ b/src/gpu/modal_export.py @@ -353,7 +353,7 @@ def stage(event: str) -> None: @app.function( cpu=8, memory=32 * 1024, - timeout=5 * 3600, + timeout=12 * 3600, volumes={**CHECKPOINT_VOLUMES, **DATASET_VOLUMES, "/outputs": MODELS_VOLUME}, ) def margin_model(