Ozan Oktay commited on
Commit
aa2430f
1 Parent(s): 8a61fc8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 3x3 cosine similarity of sentence embeddings obtained from input text prompts.
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