Skip to content

Commit b10a378

Browse files
craymichaelfacebook-github-bot
authored andcommitted
Add logprobs to results in service (#1628)
Summary: Logprobs are now added to results which can help better understand attribution scores. Differential Revision: D78491492
1 parent 4fc093e commit b10a378

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

captum/attr/_core/llm_attr.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ class LLMAttributionResult:
5555
token_attr: Optional[Tensor]
5656
input_tokens: List[str]
5757
output_tokens: List[str]
58+
output_probs: Optional[Tensor] = None
5859

5960
@property
6061
def seq_attr_dict(self) -> Dict[str, float]:

0 commit comments

Comments
 (0)