Visual Document Retrieval
ColPali
Safetensors
sentence-transformers
English
vidore
vidore-experimental
multi-vector
tomaarsen HF Staff commited on
Commit
b497847
·
verified ·
1 Parent(s): 78e60b0

Remove convert_to_tensor=True in MultiVectorEncoder, this parameter was removed

Browse files

Hello!

This simple PR removes `convert_to_tensor=True` in the MultiVectorEncoder `encode_query`/`encode_document`/`encode` calls, as I removed this parameter and made a Tensor output the default just prior to the release. I didn't realize these snippets still used this parameter, my apologies!

- Tom Aarsen

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -84,8 +84,8 @@ images = [
84
  "https://huggingface.co/datasets/sentence-transformers/example-documents/resolve/main/doc4.jpg",
85
  ]
86
 
87
- query_embeddings = model.encode_query(queries, convert_to_tensor=True)
88
- document_embeddings = model.encode_document(images, convert_to_tensor=True)
89
  print(f"Query 0 shape: {tuple(query_embeddings[0].shape)}")
90
  print(f"Document 0 shape: {tuple(document_embeddings[0].shape)}")
91
  # Query 0 shape: (28, 128)
 
84
  "https://huggingface.co/datasets/sentence-transformers/example-documents/resolve/main/doc4.jpg",
85
  ]
86
 
87
+ query_embeddings = model.encode_query(queries)
88
+ document_embeddings = model.encode_document(images)
89
  print(f"Query 0 shape: {tuple(query_embeddings[0].shape)}")
90
  print(f"Document 0 shape: {tuple(document_embeddings[0].shape)}")
91
  # Query 0 shape: (28, 128)