Ozan Oktay
commited on
Commit
•
aa2430f
1
Parent(s):
8a61fc8
Update README.md
Browse files
README.md
CHANGED
@@ -77,7 +77,7 @@ tokenizer_output = tokenizer.batch_encode_plus(batch_text_or_text_pairs=text_pro
|
|
77 |
embeddings = model.get_projected_text_embeddings(input_ids=tokenizer_output.input_ids,
|
78 |
attention_mask=tokenizer_output.attention_mask)
|
79 |
|
80 |
-
# Compute
|
81 |
sim = torch.mm(embeddings, embeddings.t())
|
82 |
```
|
83 |
|
|
|
77 |
embeddings = model.get_projected_text_embeddings(input_ids=tokenizer_output.input_ids,
|
78 |
attention_mask=tokenizer_output.attention_mask)
|
79 |
|
80 |
+
# Compute the cosine similarity of sentence embeddings obtained from input text prompts.
|
81 |
sim = torch.mm(embeddings, embeddings.t())
|
82 |
```
|
83 |
|