Fix for Remove attention module from LIT. Removed AttentionModule fro…

…m default LIT notebook layouts. PiperOrigin-RevId: 646087107
PAIR-code · Jun 24, 2024 · 71cbdba · 71cbdba
1 parent e0e35c3
commit 71cbdba
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/lit_nlp/notebook.py b/lit_nlp/notebook.py
@@ -51,7 +51,6 @@
             *layout.MODEL_PREDS_MODULES,
             modules.SalienceMapModule,
             modules.LegacySequenceSalienceModule,
-            modules.AttentionModule,
         ],
         'Analysis': [
             modules.MetricsModule,