Update README.md
Browse files
README.md
CHANGED
|
@@ -94,7 +94,6 @@ inputs = {key: value.to(device) for key, value in inputs.items()}
|
|
| 94 |
with torch.no_grad():
|
| 95 |
embeddings_dict = model(inputs)
|
| 96 |
embeddings = embeddings_dict['token_embeddings']
|
| 97 |
-
print(embeddings.shape) # Expected: [3, 32, 128] (3 texts, 32 tokens max, 128-dim embeddings)
|
| 98 |
|
| 99 |
# Define ColBERT's MaxSim similarity function
|
| 100 |
def colbert_similarity(query_emb, doc_emb):
|
|
|
|
| 94 |
with torch.no_grad():
|
| 95 |
embeddings_dict = model(inputs)
|
| 96 |
embeddings = embeddings_dict['token_embeddings']
|
|
|
|
| 97 |
|
| 98 |
# Define ColBERT's MaxSim similarity function
|
| 99 |
def colbert_similarity(query_emb, doc_emb):
|