Tom Aarsen
commited on
Commit
•
530e1f3
1
Parent(s):
0bfdc3c
Update model author to dunzhang
Browse files(Note: infgrad still works, but using dunzhang is probably less confusing)
README.md
CHANGED
@@ -5492,7 +5492,7 @@ docs = [
|
|
5492 |
]
|
5493 |
|
5494 |
# !The default dimension is 1024, if you need other dimensions, please clone the model and modify `modules.json` to replace `2_Dense_1024` with another dimension, e.g. `2_Dense_256` or `2_Dense_8192` !
|
5495 |
-
model = SentenceTransformer("
|
5496 |
query_embeddings = model.encode(queries, prompt_name=query_prompt_name)
|
5497 |
doc_embeddings = model.encode(docs)
|
5498 |
print(query_embeddings.shape, doc_embeddings.shape)
|
|
|
5492 |
]
|
5493 |
|
5494 |
# !The default dimension is 1024, if you need other dimensions, please clone the model and modify `modules.json` to replace `2_Dense_1024` with another dimension, e.g. `2_Dense_256` or `2_Dense_8192` !
|
5495 |
+
model = SentenceTransformer("dunzhang/stella_en_1.5B_v5", trust_remote_code=True).cuda()
|
5496 |
query_embeddings = model.encode(queries, prompt_name=query_prompt_name)
|
5497 |
doc_embeddings = model.encode(docs)
|
5498 |
print(query_embeddings.shape, doc_embeddings.shape)
|