Skip to content

Commit

Permalink
add e5 variants
Browse files Browse the repository at this point in the history
  • Loading branch information
fangjy6 committed Dec 2, 2024
1 parent 069a28c commit 2f1571a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion pyterrier_dr/sbert_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,8 @@ class E5(_SBertBiEncoder):

VARIANTS = {
'base' : 'intfloat/e5-base-v2',
# TODO add more here
'small': 'intfloat/e5-small-v2',
'large': 'intfloat/e5-large-v2',
}

class GTR(_SBertBiEncoder):
Expand Down

0 comments on commit 2f1571a

Please sign in to comment.