Thank you for your response. In the past, I tried BertViz, but after realizing that it could not forecast the next token like ExBert, I stopped using it. If I am wrong about that, it would suit my needs.