Enhance HVideoTool with background processing and device detection: implemented a background job system for detection and restoration tasks, updated the UI to display CUDA/CPU status, and improved device diagnostics. Documentation in README and CLAUDE.md reflects these changes.

2026-06-07 05:24:06 +03:00
parent e27dfdf518
commit cc518cc3e6
7 changed files with 497 additions and 129 deletions
@@ -118,6 +118,14 @@ class DeepMosaicsRestorer(Restorer):
            return
        if str(_VENDOR) not in sys.path:
            sys.path.insert(0, str(_VENDOR))  # so vendored `from models/util import …` resolve
+
+        # Fall back to CPU when CUDA isn't available: DeepMosaics calls `.cuda()`
+        # whenever gpu_id != "-1", which raises "Torch not compiled with CUDA enabled"
+        # on a CPU-only torch build.
+        import torch  # noqa: E402
+        if self._gpu != "-1" and not torch.cuda.is_available():
+            self._gpu = "-1"
+
        from models import loadmodel, runmodel  # type: ignore  # noqa: E402
        import util.image_processing as impro    # type: ignore  # noqa: E402