Skip to content

Commit 27d7a84

Browse files
llcourageLIT team
authored andcommitted
Remove attention module from LIT.
PiperOrigin-RevId: 645032202
1 parent 0656386 commit 27d7a84

File tree

3 files changed

+6
-358
lines changed

3 files changed

+6
-358
lines changed

lit_nlp/api/layout.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@ class LitModuleName(dtypes.EnumSerializableAsValues, enum.Enum):
3232
# keep-sorted start
3333
AnnotatedTextGoldModule = 'annotated-text-gold-module'
3434
AnnotatedTextModule = 'annotated-text-module'
35-
AttentionModule = 'attention-module'
3635
ClassificationModule = 'classification-module'
3736
ConfusionMatrixModule = 'confusion-matrix-module'
3837
CurvesModule = 'curves-module'
@@ -198,7 +197,6 @@ def to_json(self) -> dtypes.JsonDict:
198197
'Explanations': [
199198
modules.SalienceMapModule,
200199
modules.LegacySequenceSalienceModule,
201-
modules.AttentionModule,
202200
modules.FeatureAttributionModule,
203201
],
204202
'Clustering': [modules.SalienceClusteringModule],
@@ -232,7 +230,6 @@ def to_json(self) -> dtypes.JsonDict:
232230
*MODEL_PREDS_MODULES,
233231
modules.SalienceMapModule,
234232
modules.LegacySequenceSalienceModule,
235-
modules.AttentionModule,
236233
modules.FeatureAttributionModule,
237234
],
238235
'Salience Clustering': [modules.SalienceClusteringModule],

lit_nlp/client/modules/attention_module.ts

Lines changed: 0 additions & 350 deletions
This file was deleted.

0 commit comments

Comments
 (0)