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 |
+
```
|