Add logprobs to results in service

craymichael · facebook-github-bot · commit 4ed82f9b606a · 2025-07-17T08:38:13.000-07:00
Summary: Logprobs are now added to results which can help better understand attribution scores.

Differential Revision: D78491492
diff --git a/captum/attr/_core/llm_attr.py b/captum/attr/_core/llm_attr.py
@@ -55,6 +55,7 @@ class LLMAttributionResult:
     token_attr: Optional[Tensor]
     input_tokens: List[str]
     output_tokens: List[str]
+    output_probs: Tensor | None = None
 
     @property
     def seq_attr_dict(self) -> Dict[str, float]:
