update readme
Browse files
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 =
|
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 |
+
```
|