Repository navigation

24 add cross attention labels text - #60

Merged

meilame-tayebjee merged 22 commits into

mainfrom

24-add-cross-attention-labels-text

Jan 27, 2026

Merged

24 add cross attention labels text#60
meilame-tayebjee merged 22 commits into
mainfrom
24-add-cross-attention-labels-text

Commits on Jan 26, 2026
- feat!(label attention): enable label attention
  meilame-tayebjee
  committed
  601fa46
- fix(load): restore LabelAttentionConfig object from loaded dict
  meilame-tayebjee
  committed
  30ef8af
- test(label attention): add a test_pipeline with label attention activated
  meilame-tayebjee
  committed
  0a1880b
- feat(explainability): add new expl. pipe. with label attention
  meilame-tayebjee
  committed
  a2fe33e
- fix typo
  
  meilame-tayebjee
  and
  Copilot
  authored
  3d034f4
- fix docstring in TextEmbedder forward
  
  meilame-tayebjee
  and
  Copilot
  authored
  ec6742c
- fix: convert to LabelAttentionConfig object when dict
  
  meilame-tayebjee
  and
  Copilot
  authored
  f991b6b
- chore: replace type checking with isinstance
  
  meilame-tayebjee
  and
  Copilot
  authored
  525b482
- chore: better dict handling in TextEmbedder forward output
  
  meilame-tayebjee
  and
  Copilot
  authored
  2374df8
Commits on Jan 27, 2026
- fix: ensure label_indices uses correct device and dtype
  
  Copilot
  and
  meilame-tayebjee
  committed
  d266572
- Add validation for LabelAttentionClassifier head configuration
  
  Copilot
  and
  meilame-tayebjee
  committed
  4aada37
- Improve validation to follow TextEmbedder pattern and clarify error messages
  
  Copilot
  and
  meilame-tayebjee
  committed
  d516e6b
- Apply attention mask in LabelAttentionClassifier cross-attention
  
  Copilot
  and
  meilame-tayebjee
  committed
  87d672f
- Fix trailing whitespace in attention matrix computation
  
  Copilot
  and
  meilame-tayebjee
  committed
  7a988c3
- doc: fix docstring on shape of label attention matrix
  
  meilame-tayebjee
  and
  Copilot
  authored
  70b79c9
- Add assertions for label attention attributions in tests
  
  Copilot
  and
  meilame-tayebjee
  committed
  0558f97
- Update _get_sentence_embedding return type annotation and docstring
  
  Copilot
  and
  meilame-tayebjee
  committed
  44d9345
- Fix early returns to match dictionary return type
  
  Copilot
  and
  meilame-tayebjee
  committed
  86a0715
- Initialize label_attention_matrix to None before text_embedder branch
  
  Copilot
  and
  meilame-tayebjee
  committed
  a721397
- Fix return type annotation for TextClassificationModel.forward
  
  Copilot
  and
  meilame-tayebjee
  committed
  798ff8c
- Address code review feedback: fix trailing whitespace and NameError
  
  Copilot
  and
  meilame-tayebjee
  committed
  1bbff15
- chore: disable gqa for label attention for simpl.
  meilame-tayebjee
  committed
  308fddc

Commits

Commits on Jan 26, 2026

Commits on Jan 27, 2026