Address PR review feedback

2026-05-12 20:45:59 +02:00
parent a54e64c72c
commit 7bc79783d4
1 changed files with 1 additions and 1 deletions
@@ -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