I’m not familiar with the linked repository, but could you try to use Captum 20?
I think these methods are also implemented there.
CC @Narine_Kokhlikyan who will know, if Captum can be used for this use case.