Skip to content

Commit

Permalink
fix: fix bug-causing spelling error in function name of e5-mistral-in…
Browse files Browse the repository at this point in the history
…struct (#1106)

found bug
  • Loading branch information
anpalmak2003 authored Jul 22, 2024
1 parent aa32a26 commit 2759cc1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mteb/models/e5_instruct.py
Original file line number Diff line number Diff line change
Expand Up @@ -193,7 +193,7 @@ def last_token_pool(last_hidden_states: Tensor, attention_mask: Tensor) -> Tenso
sequence_lengths,
]

def get_embbeding_from_output(
def get_embedding_from_output(
self, output: ModelOutput, batch_dict: BatchEncoding
) -> torch.Tensor:
return self.last_token_pool(
Expand Down

0 comments on commit 2759cc1

Please sign in to comment.