Remove convert_to_tensor=True in MultiVectorEncoder, this parameter was removed
#13
by tomaarsen HF Staff - opened
README.md
CHANGED
|
@@ -83,8 +83,8 @@ images = [
|
|
| 83 |
"https://huggingface.co/datasets/sentence-transformers/example-documents/resolve/main/doc4.jpg",
|
| 84 |
]
|
| 85 |
|
| 86 |
-
query_embeddings = model.encode_query(queries
|
| 87 |
-
document_embeddings = model.encode_document(images
|
| 88 |
print(f"Query 0 shape: {tuple(query_embeddings[0].shape)}")
|
| 89 |
print(f"Document 0 shape: {tuple(document_embeddings[0].shape)}")
|
| 90 |
# Query 0 shape: (23, 128)
|
|
|
|
| 83 |
"https://huggingface.co/datasets/sentence-transformers/example-documents/resolve/main/doc4.jpg",
|
| 84 |
]
|
| 85 |
|
| 86 |
+
query_embeddings = model.encode_query(queries)
|
| 87 |
+
document_embeddings = model.encode_document(images)
|
| 88 |
print(f"Query 0 shape: {tuple(query_embeddings[0].shape)}")
|
| 89 |
print(f"Document 0 shape: {tuple(document_embeddings[0].shape)}")
|
| 90 |
# Query 0 shape: (23, 128)
|