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)

Files changed (1) hide show
  1. README.md +1 -1
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("infgrad/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)
 
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)