Revert "Fix extract_features_value_attention and add _get_all_blocks"

This reverts commit f55b5f107d6b313b89520b4fc260a34207adc461.

app.py

@@ -36,15 +36,7 @@ DEFAULT_VARIANT = "TIPS v2 — L/14"
 
 def _device():
     return torch.device("cuda" if torch.cuda.is_available() else "cpu")
-def _get_all_blocks(model_image):
-        if hasattr(model_image, "chunked_blocks") and model_image.chunked_blocks:
-                    blocks = []
-                    for chunk in model_image.blocks:
-                                    for blk in chunk:
-                                                        if not isinstance(blk, torch.nn.Identity):
-                                                                                blocks.append(blk)
-                    return blocks
-                return list(model_image.blocks)
+
 
 # ── Pascal Context (59 classes) ─────────────────────────────────────────────