Views
No views yet
| Model | ViDoRe V2 (nDCG@5) | MMEB VisDoc (ndcg_linear@5) |
|---|---|---|
| ApsaraStackMaaS/EvoQwen2.5-VL-Retriever-3B-v1 | 63.00 | 75.96 |
| ApsaraStackMaaS/EvoQwen2.5-VL-Retriever-7B-v1 | 65.24 | 77.10 |
1
2 import torch
3 from PIL import Image
4 from transformers.utils.import_utils import is_flash_attn_2_available
5[object Object]
6[object Object]
7[object Object]
8[object Object]
9 [object Object]
10 images = [
11 Image.new("RGB", (128, 128), color="white"),
12 Image.new("RGB", (64, 32), color="black"),
13 ]
14 queries = [
15 "Is attention really all you need?",
16 "What is the amount of bananas farmed in Salvador?",
17 ]
18 [object Object]
19 batch_images = processor.process_images(images).to(model.device)
20 batch_queries = processor.process_queries(queries).to(model.device)
21 [object Object]
22 with torch.no_grad():
23 image_embeddings = model(**batch_images)
24 query_embeddings = model(**batch_queries)
25[object Object]1
2[object Object]
3[object Object]
4[object Object]
5
6 [object Object]