/attentionAnalyzer

Analyze and understand what attention heads a transformer-based language model is paying attention to.

Primary LanguagePythonMIT LicenseMIT

Stargazers