Samoed commited on
Commit
afc3533
·
1 Parent(s): 2fff2f2

update readme

Browse files
Files changed (1) hide show
  1. README.md +4 -16
README.md CHANGED
@@ -96,24 +96,12 @@ model-index:
96
  pipeline_tag: feature-extraction
97
  tags:
98
  - mteb
 
99
  ---
100
 
101
  ```python
102
- import torch.nn as nn
103
  from sentence_transformers import SentenceTransformer
104
- from modeling_adaptor import MixtureOfAdaptors
105
- class CustomSentenceTransformer(nn.Module):
106
- def __init__(self, output_dim=1536):
107
- super(CustomSentenceTransformer, self).__init__()
108
- self.model = SentenceTransformer('IEITYuan/Yuan-embedding-1.0', trust_remote_code=True)
109
- adaptor = MixtureOfAdaptors(5, 1792)
110
- adaptor.load_state_dict(torch.load(f"yuan-adaptors.pth"))
111
- self.model.add_module('adaptor', adaptor)
112
- self.output_dim = output_dim
113
-
114
- def encode(self, sentences, **kwargs):
115
- embeddings = self.model.encode(sentences, **kwargs)
116
- return embeddings[:, :self.output_dim]
117
 
118
- model = CustomSentenceTransformer(output_dim=1536)
119
- model.encode(['text'])
 
 
96
  pipeline_tag: feature-extraction
97
  tags:
98
  - mteb
99
+ - sentence-transformers
100
  ---
101
 
102
  ```python
 
103
  from sentence_transformers import SentenceTransformer
 
 
 
 
 
 
 
 
 
 
 
 
 
104
 
105
+ model = SentenceTransformer("OpenSearch-AI/Ops-MoA-Conan-embedding-v1")
106
+ model.encode(['text'])
107
+ ```