Upload folder using huggingface_hub

explainability.py

@@ -148,9 +148,17 @@ class ExplainabilityEngine:
                 last_hidden_state = vision_outputs.last_hidden_state
                 
                 # Text Features (Prompts, Dim)
+                # Text Features (Prompts, Dim)
+                # FIX: Robustly handle attention_mask (some processors don't return it for text-only inputs if irrelevant)
+                text_inputs_ids = inputs["input_ids"]
+                text_attention_mask = inputs.get("attention_mask")
+                
+                if text_attention_mask is None:
+                    text_attention_mask = torch.ones_like(text_inputs_ids)
+
                 text_outputs = self.model.text_model(
-                    input_ids=inputs["input_ids"],
-                    attention_mask=inputs["attention_mask"]
+                    input_ids=text_inputs_ids,
+                    attention_mask=text_attention_mask
                 )
                 text_embeds = text_outputs.pooler_output
                 text_embeds = text_embeds / text_embeds.norm(p=2, dim=-1, keepdim=True)