horde-engine 3.1.1__tar.gz → 3.2.2__tar.gz

{horde_engine-3.1.1 → horde_engine-3.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: horde-engine
-Version: 3.1.1
+Version: 3.2.2
 Summary: A wrapper around ComfyUI to allow use by AI Horde.
 Project-URL: Homepage, https://github.com/Haidra-Org/hordelib
 Project-URL: Bug Tracker, https://github.com/Haidra-Org/hordelib/issues
@@ -679,17 +679,15 @@ Requires-Dist: aiohttp>=3.11.8
 Requires-Dist: alembic
 Requires-Dist: av>=16.0.0
 Requires-Dist: blake3
-Requires-Dist: clip-anytorch>=2.6.0
 Requires-Dist: comfy-aimdo==0.4.9
 Requires-Dist: comfy-kitchen==0.2.10
 Requires-Dist: comfyui-embedded-docs==0.5.3
 Requires-Dist: comfyui-frontend-package==1.45.15
 Requires-Dist: comfyui-workflow-templates==0.9.98
-Requires-Dist: diffusers>=0.25.0
 Requires-Dist: distro>=1.9.0
 Requires-Dist: einops
-Requires-Dist: fairscale>=0.4.13
 Requires-Dist: filelock
+Requires-Dist: ftfy
 Requires-Dist: fuzzywuzzy>=0.18.0
 Requires-Dist: gitpython>=3.1.45
 Requires-Dist: horde-model-reference<6.0.0,>=5.1.1
@@ -697,11 +695,9 @@ Requires-Dist: horde-sdk>=0.20.1
 Requires-Dist: kornia>=0.7.1
 Requires-Dist: logfire[system-metrics]==4.14.2
 Requires-Dist: loguru>=0.7.3
-Requires-Dist: lpips>=0.1.4
 Requires-Dist: matplotlib>=3.10.9
 Requires-Dist: numpy>=1.25.0
 Requires-Dist: omegaconf>=2.3.0
-Requires-Dist: open-clip-torch>=2.32.0
 Requires-Dist: opencv-contrib-python>=4.11.0.86
 Requires-Dist: opencv-python>=4.11.0.86
 Requires-Dist: opentelemetry-instrumentation-aiohttp-client>=0.59b0
@@ -712,24 +708,18 @@ Requires-Dist: pydantic-settings~=2.0
 Requires-Dist: pydantic~=2.0
 Requires-Dist: pynvml>=13.0.1
 Requires-Dist: python-dotenv>=1.2.1
-Requires-Dist: pytorch-lightning>=2.5.6
 Requires-Dist: pyyaml
 Requires-Dist: qrcode==7.4.2
-Requires-Dist: rembg[cpu]>=2.0.67
 Requires-Dist: requests
 Requires-Dist: safetensors>=0.4.2
 Requires-Dist: scikit-image>=0.25.2
 Requires-Dist: scipy
 Requires-Dist: sentencepiece
 Requires-Dist: simpleeval>=1.0.0
-Requires-Dist: soundfile>=0.13.1
-Requires-Dist: spandrel-extra-arches>=0.2.0
 Requires-Dist: spandrel>=0.4.1
 Requires-Dist: sqlalchemy>=2.0.0
 Requires-Dist: strenum>=0.4.15
-Requires-Dist: timm==0.9.16
 Requires-Dist: tokenizers>=0.13.3
-Requires-Dist: torchaudio<2.13,>=2.9.0
 Requires-Dist: torchdiffeq>=0.2.5
 Requires-Dist: torchsde
 Requires-Dist: torchvision<0.28,>=0.24.0
@@ -739,14 +729,32 @@ Requires-Dist: typing-extensions>=4.15.0
 Requires-Dist: unidecode>=1.4.0
 Requires-Dist: yapf>=0.43.0
 Requires-Dist: yarl>=1.18.0
+Provides-Extra: controlnet
+Requires-Dist: albumentations; extra == 'controlnet'
+Requires-Dist: fvcore; extra == 'controlnet'
+Requires-Dist: mediapipe>=0.10; extra == 'controlnet'
+Requires-Dist: onnxruntime>=1.16; extra == 'controlnet'
+Requires-Dist: scikit-learn; extra == 'controlnet'
+Requires-Dist: trimesh[easy]; extra == 'controlnet'
+Requires-Dist: yacs; extra == 'controlnet'
 Provides-Extra: cpu
 Requires-Dist: torch; extra == 'cpu'
+Requires-Dist: torchvision; extra == 'cpu'
 Provides-Extra: cu126
 Requires-Dist: torch; extra == 'cu126'
+Requires-Dist: torchvision; extra == 'cu126'
 Provides-Extra: cu130
 Requires-Dist: torch; extra == 'cu130'
+Requires-Dist: torchvision; extra == 'cu130'
 Provides-Extra: cu132
 Requires-Dist: torch; extra == 'cu132'
+Requires-Dist: torchvision; extra == 'cu132'
+Provides-Extra: layerdiffuse
+Requires-Dist: diffusers>=0.25.0; extra == 'layerdiffuse'
+Provides-Extra: rembg
+Requires-Dist: rembg[cpu]>=2.0.67; extra == 'rembg'
+Provides-Extra: upscale-extra
+Requires-Dist: spandrel-extra-arches>=0.2.0; extra == 'upscale-extra'
 Description-Content-Type: text/markdown
 
 # horde-engine
@@ -836,29 +844,36 @@ uv sync   # installs the latest torch (2.12.0, the CUDA 12.6 build) -- see "Choo
 
 ### Choosing a PyTorch build
 
-`horde-engine` does not pin `torch`: it declares `torchvision`/`torchaudio` as version *ranges*, and
-torchvision pins the exact matching torch in its metadata, so installing the library pulls the latest
-supported torch (**2.12.0** -> torchvision 0.27.0 / torchaudio 2.11.0; torchaudio has no 2.12 release).
-You only pick the *build* (which CUDA/ROCm wheels) for your hardware.
+`horde-engine` does not pin `torch`: it declares `torchvision` as a version *range*, and torchvision
+pins the exact matching torch in its metadata, so installing the library pulls the latest supported
+torch (**2.12.0** -> torchvision 0.27.0). You only pick the *build* (which CUDA/ROCm wheels) for your
+hardware. `torch` and `torchvision` are both routed per-build in `[tool.uv.sources]` so they always
+share one CUDA build (a torchvision left on generic PyPI would pull a different CUDA build and break the
+install).
 
-For development, `uv sync` installs the **CUDA 12.6** build of torch 2.12.0 (the index hordelib routes
-to in `[tool.uv.sources]`). It runs on any CUDA 12.6+ driver and, via NVIDIA driver
-backward-compatibility, on CUDA 13 drivers too -- the broadest single build. To use a different build,
-override it ad-hoc after syncing:
+`torchaudio` is **not** installed by default: it is the only package in the trio with no `+cu132` wheel,
+and audio generation is currently unsupported. ComfyUI's eager `import torchaudio` is satisfied by a
+lazy stub (`hordelib.utils.torch_build`) when it is absent; audio operations raise a clear error only if
+actually used. Install a matching torchaudio ad-hoc if you need audio (cu126/cu130/cpu only).
+
+For development, `uv sync --extra cu126` installs the **CUDA 12.6** build of torch 2.12.0. It runs on any
+CUDA 12.6+ driver and, via NVIDIA driver backward-compatibility, on CUDA 13 drivers too -- the broadest
+single build. Pick a different build with the matching extra:
 
 ```bash
-uv sync
-# then, only if you want a build other than the cu126 default:
-UV_TORCH_BACKEND=auto uv pip install torch torchvision torchaudio                # auto-detect your GPU
-uv pip install torch --extra-index-url https://download.pytorch.org/whl/cu130    # exact CUDA 13 build
+uv sync --extra cu132   # exact CUDA 13.2 build
+uv sync --extra cu130   # CUDA 13.0/13.1 build (also runs on a 13.2 driver)
+uv sync --extra cu126   # CUDA 12.6+ build (also runs on CUDA 13 drivers)
+uv sync --extra cpu     # no GPU; ~100x slower, testing only
 ```
 
-With `pip`, select the index with `--extra-index-url`:
+With `pip`, select the index with `--extra-index-url` (install `torch` **and** `torchvision` from the
+same index):
 
 | Hardware | index | Notes |
 |---|---|---|
-| NVIDIA, CUDA 13+ driver | `cu130` | torch 2.12.0; a 13.0 build covers any 13.x driver (incl. 13.2). |
-| NVIDIA, CUDA 13.2 (exact) | `cu132` | torch 2.12.0; `cu130` also runs on a 13.2 driver. |
+| NVIDIA, CUDA 13.2 (exact) | `cu132` | torch 2.12.0; no torchaudio wheel (audio unavailable on this build). |
+| NVIDIA, CUDA 13.0/13.1 driver | `cu130` | torch 2.12.0; also runs on a 13.2 driver. |
 | NVIDIA, CUDA 12.6+ driver | `cu126` | torch 2.12.0; the only CUDA-12 build of 2.12.0. |
 | NVIDIA, CUDA 12.x, older torch | `cu128` | No 2.12.0 wheel; caps at torch **2.11.0** (add `"torch==2.11.0"`). |
 | AMD (Linux only) | `rocm6.4` | torch **2.9.1 only** (see note below). |
@@ -866,7 +881,7 @@ With `pip`, select the index with `--extra-index-url`:
 
 > **ROCm:** only torch 2.9.1 (ROCm 6.4) is currently installable. PyTorch publishes torch 2.10-2.12 on
 > rocm7.x but not the matching `pytorch-triton-rocm` (3.6.0/3.7.0) those wheels hard-depend on, so they
-> cannot be resolved yet. Install it ad-hoc: `uv pip install torch==2.9.1 torchvision torchaudio
+> cannot be resolved yet. Install it ad-hoc: `uv pip install torch==2.9.1 torchvision
 > pytorch-triton-rocm --extra-index-url https://download.pytorch.org/whl/rocm6.4`.
 
 ---

{horde_engine-3.1.1 → horde_engine-3.2.2}/README.md

@@ -85,29 +85,36 @@ uv sync   # installs the latest torch (2.12.0, the CUDA 12.6 build) -- see "Choo
 
 ### Choosing a PyTorch build
 
-`horde-engine` does not pin `torch`: it declares `torchvision`/`torchaudio` as version *ranges*, and
-torchvision pins the exact matching torch in its metadata, so installing the library pulls the latest
-supported torch (**2.12.0** -> torchvision 0.27.0 / torchaudio 2.11.0; torchaudio has no 2.12 release).
-You only pick the *build* (which CUDA/ROCm wheels) for your hardware.
-
-For development, `uv sync` installs the **CUDA 12.6** build of torch 2.12.0 (the index hordelib routes
-to in `[tool.uv.sources]`). It runs on any CUDA 12.6+ driver and, via NVIDIA driver
-backward-compatibility, on CUDA 13 drivers too -- the broadest single build. To use a different build,
-override it ad-hoc after syncing:
+`horde-engine` does not pin `torch`: it declares `torchvision` as a version *range*, and torchvision
+pins the exact matching torch in its metadata, so installing the library pulls the latest supported
+torch (**2.12.0** -> torchvision 0.27.0). You only pick the *build* (which CUDA/ROCm wheels) for your
+hardware. `torch` and `torchvision` are both routed per-build in `[tool.uv.sources]` so they always
+share one CUDA build (a torchvision left on generic PyPI would pull a different CUDA build and break the
+install).
+
+`torchaudio` is **not** installed by default: it is the only package in the trio with no `+cu132` wheel,
+and audio generation is currently unsupported. ComfyUI's eager `import torchaudio` is satisfied by a
+lazy stub (`hordelib.utils.torch_build`) when it is absent; audio operations raise a clear error only if
+actually used. Install a matching torchaudio ad-hoc if you need audio (cu126/cu130/cpu only).
+
+For development, `uv sync --extra cu126` installs the **CUDA 12.6** build of torch 2.12.0. It runs on any
+CUDA 12.6+ driver and, via NVIDIA driver backward-compatibility, on CUDA 13 drivers too -- the broadest
+single build. Pick a different build with the matching extra:
 
 ```bash
-uv sync
-# then, only if you want a build other than the cu126 default:
-UV_TORCH_BACKEND=auto uv pip install torch torchvision torchaudio                # auto-detect your GPU
-uv pip install torch --extra-index-url https://download.pytorch.org/whl/cu130    # exact CUDA 13 build
+uv sync --extra cu132   # exact CUDA 13.2 build
+uv sync --extra cu130   # CUDA 13.0/13.1 build (also runs on a 13.2 driver)
+uv sync --extra cu126   # CUDA 12.6+ build (also runs on CUDA 13 drivers)
+uv sync --extra cpu     # no GPU; ~100x slower, testing only
 ```
 
-With `pip`, select the index with `--extra-index-url`:
+With `pip`, select the index with `--extra-index-url` (install `torch` **and** `torchvision` from the
+same index):
 
 | Hardware | index | Notes |
 |---|---|---|
-| NVIDIA, CUDA 13+ driver | `cu130` | torch 2.12.0; a 13.0 build covers any 13.x driver (incl. 13.2). |
-| NVIDIA, CUDA 13.2 (exact) | `cu132` | torch 2.12.0; `cu130` also runs on a 13.2 driver. |
+| NVIDIA, CUDA 13.2 (exact) | `cu132` | torch 2.12.0; no torchaudio wheel (audio unavailable on this build). |
+| NVIDIA, CUDA 13.0/13.1 driver | `cu130` | torch 2.12.0; also runs on a 13.2 driver. |
 | NVIDIA, CUDA 12.6+ driver | `cu126` | torch 2.12.0; the only CUDA-12 build of 2.12.0. |
 | NVIDIA, CUDA 12.x, older torch | `cu128` | No 2.12.0 wheel; caps at torch **2.11.0** (add `"torch==2.11.0"`). |
 | AMD (Linux only) | `rocm6.4` | torch **2.9.1 only** (see note below). |
@@ -115,7 +122,7 @@ With `pip`, select the index with `--extra-index-url`:
 
 > **ROCm:** only torch 2.9.1 (ROCm 6.4) is currently installable. PyTorch publishes torch 2.10-2.12 on
 > rocm7.x but not the matching `pytorch-triton-rocm` (3.6.0/3.7.0) those wheels hard-depend on, so they
-> cannot be resolved yet. Install it ad-hoc: `uv pip install torch==2.9.1 torchvision torchaudio
+> cannot be resolved yet. Install it ad-hoc: `uv pip install torch==2.9.1 torchvision
 > pytorch-triton-rocm --extra-index-url https://download.pytorch.org/whl/rocm6.4`.
 
 ---

{horde_engine-3.1.1 → horde_engine-3.2.2}/hordelib/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '3.1.1'
-__version_tuple__ = version_tuple = (3, 1, 1)
+__version__ = version = '3.2.2'
+__version_tuple__ = version_tuple = (3, 2, 2)
 
 __commit_id__ = commit_id = None

{horde_engine-3.1.1 → horde_engine-3.2.2}/hordelib/api.py

@@ -31,6 +31,14 @@ from hordelib.feature_impact import (
     get_baseline_burden,
     get_feature_impact_registry,
 )
+from hordelib.feature_requirements import (
+    FeatureRequirement,
+    available_features,
+    feature_available,
+    get_feature_requirement,
+    get_feature_requirement_registry,
+    missing_packages,
+)
 from hordelib.horde import (
     HordeLib,
     ProgressReport,
@@ -58,6 +66,10 @@ from hordelib.shared_model_manager import SharedModelManager
 from hordelib.utils.ioredirect import ComfyUIProgress, ComfyUIProgressUnit
 from hordelib.utils.logger import HordeLog
 from hordelib.utils.torch_memory import (
+    AcceleratorInfo,
+    AcceleratorKind,
+    clear_accelerator_cache,
+    enumerate_accelerators,
     get_free_ram_mb,
     get_torch_free_vram_mb,
     get_torch_total_vram_mb,
@@ -66,6 +78,8 @@ from hordelib.utils.torch_memory import (
 
 __all__ = [
     "FEATURE_KIND",
+    "AcceleratorInfo",
+    "AcceleratorKind",
     "BaselineBurden",
     "BurdenEstimate",
     "CalibrationSample",
@@ -75,6 +89,7 @@ __all__ = [
     "DownloadTrigger",
     "FeatureImpact",
     "FeatureImpactRegistry",
+    "FeatureRequirement",
     "ExecutionBackend",
     "FacefixPayload",
     "HordeLib",
@@ -96,10 +111,17 @@ __all__ = [
     "StripBackgroundPayload",
     "UpscalePayload",
     "VRAMStats",
+    "available_features",
     "classify_post_processor",
+    "clear_accelerator_cache",
+    "enumerate_accelerators",
     "estimate_job_burden",
+    "feature_available",
     "get_baseline_burden",
     "get_feature_impact_registry",
+    "get_feature_requirement",
+    "get_feature_requirement_registry",
+    "missing_packages",
     "get_free_ram_mb",
     "get_metrics_collector",
     "get_torch_free_vram_mb",

{horde_engine-3.1.1 → horde_engine-3.2.2}/hordelib/comfy_horde.py

@@ -32,6 +32,7 @@ from hordelib.utils.ioredirect import ComfyUIProgress, OutputCollector
 from hordelib.config_path import get_hordelib_path
 from hordelib.execution import comfy_patches
 from hordelib.execution.graph_utils import GraphDict
+from hordelib.utils.torch_memory import clear_accelerator_cache
 
 # Note It may not be abundantly clear with no context what is going on below, and I will attempt to clarify:
 #
@@ -264,6 +265,13 @@ def do_comfy_import(
     if extra_comfyui_args is not None:
         sys.argv.extend(extra_comfyui_args)
 
+    # ComfyUI imports torchaudio eagerly in several node modules loaded just below. torchaudio is an
+    # optional, non-default dependency (no cu132 wheel; audio unsupported), so install a lazy stub when
+    # it is absent, BEFORE comfy is imported, so those imports succeed for image/video work.
+    from hordelib.utils.torch_build import ensure_torchaudio_importable
+
+    ensure_torchaudio_importable()
+
     # Note these imports are intentionally somewhat obfuscated as a reminder to other modules
     # that they should never call through this module into comfy directly. All calls into
     # comfy should be abstracted through functions in this module.
@@ -367,11 +375,9 @@ def do_comfy_import(
 
 
 def clear_gc_and_torch_cache() -> None:
-    """Clear the garbage collector and the PyTorch cache."""
+    """Clear the garbage collector and the active backend's device cache."""
     gc.collect()
-    from torch.cuda import empty_cache
-
-    empty_cache()
+    clear_accelerator_cache()
 
 
 def pin_models_in_vram() -> bool:
@@ -1072,7 +1078,7 @@ class Comfy_Horde:
             _mc.record_phase("pipeline_validate", _validate_seconds)
             _mc.record_phase("pipeline_execute", _execute_seconds)
             _mc.record_phase("pipeline_finalize", time.perf_counter() - _t_post_execute)
-        except Exception:  # noqa: BLE001 - instrumentation must never break a run
+        except Exception:
             pass
 
         # Record pipeline duration

horde_engine-3.2.2/hordelib/feature_requirements.py

@@ -0,0 +1,115 @@
+"""Which optional features are actually installable in this environment.
+
+Some hordelib features depend on heavy, platform-constrained native packages that are not
+part of the lean base install (they have no wheels for some accelerators, e.g. Intel XPU /
+Apple MPS / Ascend, so forcing them into base would block a base install there). Those
+packages live in ``horde-engine`` feature extras (see ``[project.optional-dependencies]``)
+and the feature's call site guards its import via :mod:`hordelib.utils.optional_deps`.
+
+This module is the single, typed source of truth mapping each such feature to the extra and
+the import-names that back it, plus a runtime probe (:func:`feature_available`) so a consumer
+(notably the worker) can decide what to advertise *before* a job arrives rather than failing
+mid-job. It is the one place to edit when ComfyUI's (or a node's) backend support changes.
+
+Features with no entry here have no optional dependency and are always considered available
+(core inference, the safety classifier, ESRGAN upscalers and the CodeFormer/GFPGAN face
+fixers are all pure PyTorch and run on every backend). Detection only: this module never
+installs anything; extras are resolved at lock/sync time by the packaging tool.
+
+This module is comfy-free and import-safe before ``hordelib.initialise()``; it is part of the
+public API surface re-exported from :mod:`hordelib.api`.
+"""
+
+import importlib.util
+
+from pydantic import BaseModel
+
+from hordelib.feature_impact import FEATURE_KIND
+
+__all__ = [
+    "FeatureRequirement",
+    "available_features",
+    "feature_available",
+    "get_feature_requirement",
+    "get_feature_requirement_registry",
+    "missing_packages",
+]
+
+
+class FeatureRequirement(BaseModel):
+    """The optional packages one feature needs, and the extra that provides them."""
+
+    feature: FEATURE_KIND
+    packages: tuple[str, ...]
+    """Top-level import names that must ALL be importable for the feature to be available."""
+    extra: str
+    """The ``horde-engine`` extra that installs ``packages`` (e.g. ``rembg`` -> ``[rembg]``)."""
+    label: str
+    """Human-readable feature name, used in "install this extra" messages."""
+
+
+# Only the backend-constrained features appear here. controlnet gates on onnxruntime alone: of the
+# preprocessors hordelib exposes as horde control_types (CONTROLNET_IMAGE_PREPROCESSOR_MAP), only
+# Openpose (DWPose) needs a blocker dep, and that dep is onnxruntime. mediapipe backs only
+# mediapipe_face / mesh_graphormer, which are not horde control_types, so it is shipped in the
+# `controlnet` extra for node parity but is not required to serve controlnet jobs.
+_REQUIREMENT_SEEDS: list[FeatureRequirement] = [
+    FeatureRequirement(
+        feature=FEATURE_KIND.strip_background,
+        packages=("rembg",),
+        extra="rembg",
+        label="strip_background (background removal)",
+    ),
+    FeatureRequirement(
+        feature=FEATURE_KIND.controlnet,
+        packages=("onnxruntime",),
+        extra="controlnet",
+        label="controlnet preprocessing (annotators)",
+    ),
+]
+
+_REGISTRY: dict[FEATURE_KIND, FeatureRequirement] = {req.feature: req for req in _REQUIREMENT_SEEDS}
+
+
+def _is_importable(package: str) -> bool:
+    """Return whether *package* can be found in the environment, never raising.
+
+    ``find_spec`` can raise (e.g. ``ModuleNotFoundError`` for a missing parent, or ``ValueError``)
+    rather than returning ``None``; for a capability probe any failure means "not available".
+    """
+    try:
+        return importlib.util.find_spec(package) is not None
+    except (ImportError, ValueError):
+        return False
+
+
+def get_feature_requirement_registry() -> dict[FEATURE_KIND, FeatureRequirement]:
+    """Return the mapping of features to their optional-dependency requirements."""
+    return _REGISTRY
+
+
+def get_feature_requirement(feature: FEATURE_KIND) -> FeatureRequirement | None:
+    """Return the requirement for *feature*, or None when it has no optional dependency."""
+    return _REGISTRY.get(feature)
+
+
+def missing_packages(feature: FEATURE_KIND) -> tuple[str, ...]:
+    """Return the requirement's packages that are not importable (empty when all present)."""
+    requirement = _REGISTRY.get(feature)
+    if requirement is None:
+        return ()
+    return tuple(package for package in requirement.packages if not _is_importable(package))
+
+
+def feature_available(feature: FEATURE_KIND) -> bool:
+    """Return whether *feature* can run here.
+
+    A feature with no requirement entry is always available; otherwise every package its
+    requirement names must be importable.
+    """
+    return not missing_packages(feature)
+
+
+def available_features() -> set[FEATURE_KIND]:
+    """Return the set of features runnable in this environment (always-available ones included)."""
+    return {feature for feature in FEATURE_KIND if feature_available(feature)}

{horde_engine-3.1.1 → horde_engine-3.2.2}/hordelib/initialisation.py

@@ -26,6 +26,7 @@ def initialise(
     disable_smart_memory: bool = False,
     do_not_load_model_mangers: bool = True,
     models_not_to_force_load: list[str] | None = None,
+    reference_offline: bool | None = None,
 ):
     """Initialise hordelib. This is required before using any other hordelib functions.
 
@@ -42,6 +43,9 @@ def initialise(
             ``KNOWN_IMAGE_GENERATION_BASELINE`` members (preferred) or raw comfy class-name fragments. \
             **If this is `None`, the defaults are used.** If you wish to override the defaults, pass an empty list. \
             Defaults to None.
+        reference_offline (bool | None, optional): If True, the model reference manager reads references \
+            from local disk only and never downloads them (the caller/parent process owns downloading). \
+            If None, defers to ``HORDE_MODEL_REFERENCE_OFFLINE``. Defaults to None.
     """
     global _is_initialised
 
@@ -83,6 +87,12 @@ def initialise(
     # Modify python path to include comfyui
     set_system_path()
 
+    # Fail fast (before ComfyUI imports torch) when the installed torch/torchvision (or a hand-installed
+    # torchaudio) were built for different CUDA/CPU backends, with a message that names the fix.
+    from hordelib.utils.torch_build import verify_torch_build_consistency
+
+    verify_torch_build_consistency()
+
     import hordelib.comfy_horde
 
     hordelib.comfy_horde.do_comfy_import(
@@ -116,6 +126,9 @@ def initialise(
     # Initialise model manager
     from hordelib.shared_model_manager import SharedModelManager
 
+    if reference_offline is not None:
+        SharedModelManager._reference_offline = reference_offline
+
     SharedModelManager(do_not_load_model_mangers=do_not_load_model_mangers)
 
     sys.argv = sys_arg_bkp

{horde_engine-3.1.1 → horde_engine-3.2.2}/hordelib/installation/installer.py

@@ -24,16 +24,24 @@ def _run_git(args: list[str], cwd: Path) -> str:
     """Run a git command, returning stripped stdout.
 
     Raises:
-        GitCommandError: If git exits non-zero.
+        GitCommandError: If git is not installed/on PATH, or exits non-zero.
     """
-    result = subprocess.run(
-        ["git", *args],
-        cwd=str(cwd),
-        text=True,
-        capture_output=True,
-        encoding="utf-8",
-        errors="replace",
-    )
+    try:
+        result = subprocess.run(
+            ["git", *args],
+            cwd=str(cwd),
+            text=True,
+            capture_output=True,
+            encoding="utf-8",
+            errors="replace",
+        )
+    except FileNotFoundError as exc:
+        # A bare-name "git" that the OS cannot find raises FileNotFoundError, not a non-zero exit. Turn it
+        # into the same actionable error all callers already handle, naming why git is needed.
+        raise GitCommandError(
+            "git was not found on PATH. hordelib needs git to fetch and pin ComfyUI and its custom nodes; "
+            "install git (https://git-scm.com/downloads) and make sure it is on PATH, then retry.",
+        ) from exc
     if result.returncode != 0:
         raise GitCommandError(f"git {' '.join(args)} failed in {cwd}: {result.stderr.strip()}")
     return result.stdout.strip()

{horde_engine-3.1.1 → horde_engine-3.2.2}/hordelib/nodes/facerestore_cf/__init__.py

@@ -175,7 +175,8 @@ class FaceRestoreCFWithModel:
                         output = facerestore_model(cropped_face_t, w=codeformer_fidelity)[0]
                         restored_face = tensor2img(output, rgb2bgr=True, min_max=(-1, 1))
                     del output
-                    torch.cuda.empty_cache()
+                    if torch.cuda.is_available():
+                        torch.cuda.empty_cache()
                 except Exception as error:
                     print(f"\tFailed inference for CodeFormer: {error}", file=sys.stderr)
                     restored_face = tensor2img(cropped_face_t, rgb2bgr=True, min_max=(-1, 1))

{horde_engine-3.1.1 → horde_engine-3.2.2}/hordelib/nodes/facerestore_cf/basicsr/utils/realesrgan_utils.py

@@ -221,7 +221,8 @@ class RealESRGANer:
                 if img_mode == "L":
                     output_img = cv2.cvtColor(output_img, cv2.COLOR_BGR2GRAY)
             del output_img_t
-            torch.cuda.empty_cache()
+            if torch.cuda.is_available():
+                torch.cuda.empty_cache()
         except RuntimeError as error:
             output_img = cv2.resize(
                 self.img_pre_pad,

{horde_engine-3.1.1 → horde_engine-3.2.2}/hordelib/pipeline/context.py

@@ -5,14 +5,6 @@ from pydantic import BaseModel, ConfigDict
 
 from hordelib.pipeline.patches import ResolvedLora
 
-FLUX_BASELINES = frozenset(
-    {
-        KNOWN_IMAGE_GENERATION_BASELINE.flux_1,
-        KNOWN_IMAGE_GENERATION_BASELINE.flux_schnell,
-        KNOWN_IMAGE_GENERATION_BASELINE.flux_dev,
-    },
-)
-
 
 class PostProcessingContext(BaseModel):
     """Resolved facts for a post-processing payload: where the model lives on disk.
@@ -48,15 +40,3 @@ class ModelContext(BaseModel):
     """LoRAs that have been validated/downloaded, in application order."""
     will_load_loras: bool = False
     """Whether the model loader should prepare for LoRA application."""
-
-    @property
-    def is_flux(self) -> bool:
-        return self.baseline in FLUX_BASELINES
-
-    @property
-    def is_cascade(self) -> bool:
-        return self.baseline == KNOWN_IMAGE_GENERATION_BASELINE.stable_cascade
-
-    @property
-    def is_qwen(self) -> bool:
-        return self.baseline == KNOWN_IMAGE_GENERATION_BASELINE.qwen_image