From b5d36101e4a039c3a3502d95d6d7e024564b5160 Mon Sep 17 00:00:00 2001 From: Gabriele Sarti Date: Tue, 30 May 2023 12:51:18 +0200 Subject: [PATCH] Patch-fix new step scores arguments --- inseq/attr/step_functions.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/inseq/attr/step_functions.py b/inseq/attr/step_functions.py index e296f15d..e1c5ab81 100644 --- a/inseq/attr/step_functions.py +++ b/inseq/attr/step_functions.py @@ -252,6 +252,7 @@ def pcxmi_fn( attribution_model=attribution_model, contrast_sources=contrast_sources, contrast_target_prefixes=contrast_target_prefixes, + target_ids=target_ids, **kwargs, ) return -torch.log2(torch.div(original_probs, contrast_probs)) @@ -260,6 +261,7 @@ def pcxmi_fn( def kl_divergence_fn( attribution_model: "AttributionModel", forward_output: ModelOutput, + target_ids: TargetIdsTensor, contrast_sources: Optional[FeatureAttributionInput] = None, contrast_target_prefixes: Optional[FeatureAttributionInput] = None, top_k: int = 0,