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
|