Remove convert_to_tensor=True in MultiVectorEncoder, this parameter was removed
#6
by tomaarsen HF Staff - opened
README.md
CHANGED
|
@@ -68,8 +68,8 @@ images = [
|
|
| 68 |
"https://huggingface.co/datasets/sentence-transformers/example-documents/resolve/main/doc4.jpg",
|
| 69 |
]
|
| 70 |
|
| 71 |
-
query_embeddings = model.encode_query(queries
|
| 72 |
-
document_embeddings = model.encode_document(images
|
| 73 |
print(f"Query 0 shape: {tuple(query_embeddings[0].shape)}")
|
| 74 |
print(f"Document 0 shape: {tuple(document_embeddings[0].shape)}")
|
| 75 |
# Query 0 shape: (26, 128)
|
|
|
|
| 68 |
"https://huggingface.co/datasets/sentence-transformers/example-documents/resolve/main/doc4.jpg",
|
| 69 |
]
|
| 70 |
|
| 71 |
+
query_embeddings = model.encode_query(queries)
|
| 72 |
+
document_embeddings = model.encode_document(images)
|
| 73 |
print(f"Query 0 shape: {tuple(query_embeddings[0].shape)}")
|
| 74 |
print(f"Document 0 shape: {tuple(document_embeddings[0].shape)}")
|
| 75 |
# Query 0 shape: (26, 128)
|