diff --git a/captum/attr/_core/llm_attr.py b/captum/attr/_core/llm_attr.py index 7c124b727..90540b1bd 100644 --- 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: Optional[Tensor] = None @property def seq_attr_dict(self) -> Dict[str, float]: