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