Update README.md
Browse files
README.md
CHANGED
@@ -61,7 +61,7 @@ from PIL import Image
|
|
61 |
|
62 |
from colpali_engine.models import ColQwen2_5, ColQwen2_5_Processor
|
63 |
|
64 |
-
model = ColQwen2_5
|
65 |
"vidore/colqwen2.5-v0.1",
|
66 |
torch_dtype=torch.bfloat16,
|
67 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|
|
|
61 |
|
62 |
from colpali_engine.models import ColQwen2_5, ColQwen2_5_Processor
|
63 |
|
64 |
+
model = ColQwen2_5.from_pretrained(
|
65 |
"vidore/colqwen2.5-v0.1",
|
66 |
torch_dtype=torch.bfloat16,
|
67 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|