AQ-MedAI
/

Diver-Retriever-4B

text-generation

text-generation-inference

Model card Files Files and versions

prayerdan commited on 6 days ago

Commit

3d5956f

·

verified ·

1 Parent(s): dccc7c4

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())
-# [[0.7534257769584656, 0.1146894246339798], [0.03198453038930893, 0.6258305311203003]]
 ```

 embeddings = F.normalize(embeddings, p=2, dim=1)
 scores = (embeddings[:2] @ embeddings[2:].T)
 print(scores.tolist())
 ```