AQ-MedAI
/

Diver-Retriever-4B

text-generation

text-generation-inference

Model card Files Files and versions

prayerdan commited on 5 days ago

Commit

9be9d0f

·

verified ·

1 Parent(s): 3d5956f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -546,7 +546,7 @@ embeddings = last_token_pool(outputs.last_hidden_state, batch_dict['attention_ma
 embeddings = F.normalize(embeddings, p=2, dim=1)
 scores = (embeddings[:2] @ embeddings[2:].T)
 print(scores.tolist())
 ```

 embeddings = F.normalize(embeddings, p=2, dim=1)
 scores = (embeddings[:2] @ embeddings[2:].T)
 print(scores.tolist())
+# [[0.9319270849227905, 0.5878604054450989], [0.639923095703125, 0.7950234413146973]]
 ```