McGill-NLP
/

LLM2Vec-Mistral-7B-Instruct-v2-mntp-supervised

Model card Files Files and versions Community

vaibhavad commited on Apr 4

Commit

0801a6c

•

1 Parent(s): f2163ff

Update README.md

Files changed (1) hide show

README.md +5 -2

README.md CHANGED Viewed

@@ -55,8 +55,11 @@ q_reps_norm = torch.nn.functional.normalize(q_reps, p=2, dim=1)
 d_reps_norm = torch.nn.functional.normalize(d_reps, p=2, dim=1)
 cos_sim = torch.mm(q_reps_norm, d_reps_norm.transpose(0, 1))
-print(cos_sim.tolist())
 ```
 ## Questions

 d_reps_norm = torch.nn.functional.normalize(d_reps, p=2, dim=1)
 cos_sim = torch.mm(q_reps_norm, d_reps_norm.transpose(0, 1))
+print(cos_sim)
+"""
+tensor([[0.5486, 0.0554],
+        [0.0567, 0.5437]])
+"""
 ```
 ## Questions