bertviz for mt5-base

[preethip02]

I am trying to visualise the attention layer in the mt5-base model using bertviz but am encountering errors saying that i must specify atleast one attention arguement. The code i am using is the sample code given for encoder/decoder models, with just the model changed. Does bertviz not support mt5?