feat: cuda arch detection for shorter builds, build failure detection

1 files changed, 8 insertions, 1 deletions

diff --git a/app/backends/audiocpp/models.py b/app/backends/audiocpp/models.py
index 75bc06a..772ae8d 100644
--- a/app/backends/audiocpp/models.py
+++ b/app/backends/audiocpp/models.py
@@ -11,6 +11,11 @@ from typing import Dict, List, Optional, Set, Tuple
 from backends import common
 from . import catalog as _catalog
 
+# No-output watchdog for model downloads: huggingface_hub streams steady
+# byte progress, so 5 minutes of silence means the transfer wedged. The
+# runner kills it and reports exit 124 (see run_console_subprocess).
+DOWNLOAD_STALL_TIMEOUT = 300
+
 def _installed_display_names(audiocpp_dir: Path,
                              model_entries: Optional[List[dict]],
                              install_guidance: List[Tuple[str, str]]
@@ -133,7 +138,9 @@ def _install_models(audiocpp_dir: Path,
         try:
             rc = common.run_console_subprocess(
                 argv, cwd=str(audiocpp_dir), emit=emit, cancel=cancel,
-                on_cancel=on_cancel)
+                on_cancel=on_cancel,
+                stall_timeout=(DOWNLOAD_STALL_TIMEOUT
+                               if supports_progress else None))
         except OSError as exc:
             print(f"[WARNING] Could not run python {manager} install "
                   f"{install_id}: {exc}")