Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

prevent deepshap from being used with enformer models #6

Merged
merged 1 commit into from
Jun 26, 2024

Conversation

avantikalal
Copy link
Collaborator

Deepshap should not be used with EnformerModel/EnformerPretrainedModel classes right now, as they include custom nonlinearities that are not included in the list of nonlinearities that work with deepshap. Until a fix for this is in, I raised an error if deepshap is used with these model types. Also edited tutorials 4 and 5 to use other attribution methods.

@gokceneraslan gokceneraslan merged commit 540c9ec into main Jun 26, 2024
1 check passed
@avantikalal avantikalal deleted the tangermeme-deepshap-fix branch July 16, 2024 16:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants