From 5d257c6afc2e544d02c4025634a5650f7716b781 Mon Sep 17 00:00:00 2001 From: Peer Duensing Date: Wed, 18 Dec 2024 20:44:15 +0100 Subject: [PATCH] fix: passing parameters to internal function call --- .../explainers/text/multilabel_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/transformers_interpret/explainers/text/multilabel_classification.py b/transformers_interpret/explainers/text/multilabel_classification.py index d2fa657..b03ce79 100644 --- a/transformers_interpret/explainers/text/multilabel_classification.py +++ b/transformers_interpret/explainers/text/multilabel_classification.py @@ -157,7 +157,7 @@ def __call__( ) self.selected_index = i explainer._forward = self._forward - explainer(text, i, embedding_type) + explainer(text, i, None, embedding_type, internal_batch_size, n_steps) self.attributions.append(explainer.attributions) self.input_ids = explainer.input_ids