Update README.md
Browse files
README.md
CHANGED
@@ -8,6 +8,7 @@ tags:
|
|
8 |
- vidore-experimental
|
9 |
datasets:
|
10 |
- vidore/colpali_train_set
|
|
|
11 |
---
|
12 |
|
13 |
# ColPali: Visual Retriever based on PaliGemma-3B with ColBERT strategy
|
@@ -93,9 +94,9 @@ batch_queries = processor.process_queries(queries).to(model.device)
|
|
93 |
# Forward pass
|
94 |
with torch.no_grad():
|
95 |
image_embeddings = model(**batch_images)
|
96 |
-
|
97 |
|
98 |
-
scores = processor.score_multi_vector(
|
99 |
```
|
100 |
|
101 |
## Limitations
|
|
|
8 |
- vidore-experimental
|
9 |
datasets:
|
10 |
- vidore/colpali_train_set
|
11 |
+
new_version: vidore/colpali-v1.3-merged
|
12 |
---
|
13 |
|
14 |
# ColPali: Visual Retriever based on PaliGemma-3B with ColBERT strategy
|
|
|
94 |
# Forward pass
|
95 |
with torch.no_grad():
|
96 |
image_embeddings = model(**batch_images)
|
97 |
+
query_embeddings = model(**batch_queries)
|
98 |
|
99 |
+
scores = processor.score_multi_vector(query_embeddings, image_embeddings)
|
100 |
```
|
101 |
|
102 |
## Limitations
|