How can I visualize attention for ONMT-py

Hello

I saw this question was asked in the past but I do not see a relevant response. If there is, please do assist

How can I visualize attention or generate heatmaps in ONMT-py for translation use case?

Thank you