Update README.md
Browse files
README.md
CHANGED
@@ -107,4 +107,7 @@ for i in range(num_queries):
|
|
107 |
query_embedding = query_embeddings[i].unsqueeze(0)
|
108 |
similarities = cosine_similarity(query_embedding, doc_embeddings)
|
109 |
print(f"Similarities for Query {i+1}: {similarities.cpu().float().numpy()}")
|
110 |
-
```
|
|
|
|
|
|
|
|
107 |
query_embedding = query_embeddings[i].unsqueeze(0)
|
108 |
similarities = cosine_similarity(query_embedding, doc_embeddings)
|
109 |
print(f"Similarities for Query {i+1}: {similarities.cpu().float().numpy()}")
|
110 |
+
```
|
111 |
+
|
112 |
+
### Citation
|
113 |
+
If you find this checkpoint is helpful, please consider cite Phi3, Docmatix and our DSE work.
|