From db526b39b4038d99ebdd103ecd75c21862855910 Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Wed, 22 Apr 2026 08:33:48 +0000 Subject: [PATCH] Narrow fallback import handling Agent-Logs-Url: https://github.com/IBB666/IBB_POSE/sessions/be2c1a27-aacc-43a8-adc6-0bb4dbae296e Co-authored-by: kukulu00 <1170651+kukulu00@users.noreply.github.com> --- models/HeatmapHead.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/models/HeatmapHead.py b/models/HeatmapHead.py index 9b10b01..eb22380 100644 --- a/models/HeatmapHead.py +++ b/models/HeatmapHead.py @@ -17,7 +17,7 @@ from torch import nn try: from mmpose.registry import MODELS -except Exception as _mmpose_import_error: +except ImportError as _mmpose_import_error: MODELS = None else: _mmpose_import_error = None @@ -269,7 +269,8 @@ class _FallbackHeatmapHead(nn.Module): idx = int(key_parts[1]) except ValueError as exc: raise RuntimeError( - f"Incompatible decoder checkpoint key: {key!r}" + f"Incompatible decoder checkpoint key: {key!r}. " + "Expected a numeric final_layer index." ) from exc num_conv_layers = len(self.conv_layers) if isinstance(self.conv_layers, nn.Sequential) else 0 if idx < num_conv_layers: