Update README.md
Browse files
README.md
CHANGED
@@ -56,8 +56,8 @@ with torch.no_grad():
|
|
56 |
embeddings = model(**inputs, return_dict=True).last_hidden_state[:, 0]
|
57 |
embeddings = embeddings / embeddings.norm(dim=1, keepdim=True)
|
58 |
|
59 |
-
print(embeddings[0] @ embeddings[1]) # tensor(0.
|
60 |
-
print(embeddings[0] @ embeddings[2]) # tensor(0.
|
61 |
```
|
62 |
|
63 |
## Citation
|
|
|
56 |
embeddings = model(**inputs, return_dict=True).last_hidden_state[:, 0]
|
57 |
embeddings = embeddings / embeddings.norm(dim=1, keepdim=True)
|
58 |
|
59 |
+
print(embeddings[0] @ embeddings[1]) # tensor(0.8243)
|
60 |
+
print(embeddings[0] @ embeddings[2]) # tensor(0.5814)
|
61 |
```
|
62 |
|
63 |
## Citation
|