From 71cbdbaee0fee8e96f52cd4df7a269a0873b9259 Mon Sep 17 00:00:00 2001
From: Googler <no-reply@google.com>
Date: Mon, 24 Jun 2024 07:28:43 -0700
Subject: [PATCH] Fix for Remove attention module from LIT. Removed
 AttentionModule from default LIT notebook layouts.

PiperOrigin-RevId: 646087107
---
 lit_nlp/notebook.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/lit_nlp/notebook.py b/lit_nlp/notebook.py
index aae2eb9a..02ad0e4e 100644
--- 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,