Skip to content

Add tokenizer_kwargs and model_kwargs to explain #114

Description

@AntoninPoche

Description

Add tokenizer_kwargs and model_kwargs to AttributionExplainer.explain() and propagate to the called functions.

These parameters should be dictionaries and not an undefined **kwargs.

Motivation

We initially added a default truncation=True, but @NLP-traveller has other parameter needs. Depending on the use case, one might want to modify the default parameters used during tokenization or model forwarding. For example, truncation=True, max_length=128...

Commit to Help

  • I'm willing to help with this feature.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Fields

No fields configured for Feature.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions