Skip to content

Refactor TextEmbedder#86

Merged
meilame-tayebjee merged 4 commits intomainfrom
refactor_label_attention
Apr 20, 2026
Merged

Refactor TextEmbedder#86
meilame-tayebjee merged 4 commits intomainfrom
refactor_label_attention

Conversation

@meilame-tayebjee
Copy link
Copy Markdown
Member

Create SentenceEmbedder on top of TokenEmbedder

TokenEmbedder -> takes tokenized input, outputs. tensor of size (bs, cs, d_embed)
SequenceEMbedder -> follows the TokenEmbedder and outputs a (bs, d_embed) tensor
- moved LabelAttention as one SentenceEmbedder type
- adapt tests
@meilame-tayebjee meilame-tayebjee merged commit 997a4f8 into main Apr 20, 2026
4 checks passed
@meilame-tayebjee meilame-tayebjee deleted the refactor_label_attention branch April 20, 2026 15:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant