From 7bc79783d4d63759a16030888cf0b2ed5b6f3b22 Mon Sep 17 00:00:00 2001
From: xmarre <mmquant1@gmail.com>
Date: Tue, 12 May 2026 20:45:59 +0200
Subject: [PATCH] Address PR review feedback

---
 tide_core/patches.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tide_core/patches.py b/tide_core/patches.py
index 159f9ee..d02bcf9 100644
--- a/tide_core/patches.py
+++ b/tide_core/patches.py
@@ -97,7 +97,7 @@ class TIDEAttentionPatch:
                 try:
                     text_tokens = int(img_slice[0])
                     total_tokens = int(k.shape[2])
-                except Exception:
+                except (TypeError, ValueError, IndexError, AttributeError):
                     text_tokens = 0
                     total_tokens = 0