tattrongvu
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -12,7 +12,7 @@ base_model:
|
|
12 |
tags:
|
13 |
- vidore
|
14 |
- multimodal-embedding
|
15 |
-
library_name:
|
16 |
---
|
17 |
# ColQwen2-7B: Visual Retriever based on Qwen2-VL-7B-Instruct with ColBERT strategy
|
18 |
|
@@ -65,11 +65,11 @@ from PIL import Image
|
|
65 |
from colpali_engine.models import ColQwen2, ColQwen2Processor
|
66 |
|
67 |
model = ColQwen2.from_pretrained(
|
68 |
-
"tsystems/colqwen2-7b-v1.0",
|
69 |
torch_dtype=torch.bfloat16,
|
70 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|
71 |
).eval()
|
72 |
-
processor = ColQwen2Processor.from_pretrained("tsystems/colqwen2-7b-v1.0")
|
73 |
|
74 |
# Your inputs
|
75 |
images = [
|
|
|
12 |
tags:
|
13 |
- vidore
|
14 |
- multimodal-embedding
|
15 |
+
library_name: transformers
|
16 |
---
|
17 |
# ColQwen2-7B: Visual Retriever based on Qwen2-VL-7B-Instruct with ColBERT strategy
|
18 |
|
|
|
65 |
from colpali_engine.models import ColQwen2, ColQwen2Processor
|
66 |
|
67 |
model = ColQwen2.from_pretrained(
|
68 |
+
"tsystems/colqwen2-7b-v1.0-merged",
|
69 |
torch_dtype=torch.bfloat16,
|
70 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|
71 |
).eval()
|
72 |
+
processor = ColQwen2Processor.from_pretrained("tsystems/colqwen2-7b-v1.0-merged")
|
73 |
|
74 |
# Your inputs
|
75 |
images = [
|