hotchpotch
/

vespa-onnx-intfloat-multilingual-e5-small

Feature Extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

hotchpotch commited on Mar 26

Commit

3cf845d

•

1 Parent(s): 002fc0a

Update README.md

Files changed (1) hide show

README.md +0 -4

README.md CHANGED Viewed

@@ -48,10 +48,6 @@ model.output_names["logits"] = 0
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-def cls_pool(last_hidden_state: Tensor) -> Tensor:
-    return last_hidden_state[:, 0, :]
 def average_pool(last_hidden_state: Tensor, attention_mask: Tensor) -> Tensor:
     last_hidden = last_hidden_state.masked_fill(~attention_mask[..., None].bool(), 0.0)
     return last_hidden.sum(dim=1) / attention_mask.sum(dim=1)[..., None]

 tokenizer = AutoTokenizer.from_pretrained(model_name)
 def average_pool(last_hidden_state: Tensor, attention_mask: Tensor) -> Tensor:
     last_hidden = last_hidden_state.masked_fill(~attention_mask[..., None].bool(), 0.0)
     return last_hidden.sum(dim=1) / attention_mask.sum(dim=1)[..., None]