pawasthy commited on
Commit
a52a728
1 Parent(s): ef6db93

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -12,7 +12,7 @@ language:
12
  - nl
13
  - pt
14
  - zh
15
- license: apache-2.0
16
  library_name: transformers
17
  tags:
18
  - language
@@ -670,6 +670,7 @@ model-index:
670
  value: 0.40377
671
  - type: recall_at_5
672
  value: 0.4903
 
673
  ---
674
  # Granite-Embedding-107m-multilingual
675
 
@@ -757,7 +758,7 @@ tokenized_queries = tokenizer(input_queries, padding=True, truncation=True, retu
757
  with torch.no_grad():
758
  # Queries
759
  model_output = model(**tokenized_queries)
760
- # Perform pooling. granite-embedding-30m-english uses CLS Pooling
761
  query_embeddings = model_output[0][:, 0]
762
 
763
  # normalize the embeddings
@@ -849,4 +850,4 @@ The data used to train the base language model was filtered to remove text conta
849
  year = {2024},
850
  url = {https://arxiv.org/abs/0000.00000},
851
  }
852
- ``` -->
 
12
  - nl
13
  - pt
14
  - zh
15
+ license: apache-2.0
16
  library_name: transformers
17
  tags:
18
  - language
 
670
  value: 0.40377
671
  - type: recall_at_5
672
  value: 0.4903
673
+ pipeline_tag: sentence-similarity
674
  ---
675
  # Granite-Embedding-107m-multilingual
676
 
 
758
  with torch.no_grad():
759
  # Queries
760
  model_output = model(**tokenized_queries)
761
+ # Perform pooling. granite-embedding-107m-multilingual uses CLS Pooling
762
  query_embeddings = model_output[0][:, 0]
763
 
764
  # normalize the embeddings
 
850
  year = {2024},
851
  url = {https://arxiv.org/abs/0000.00000},
852
  }
853
+ ``` -->