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,