Update README.md
Browse files
README.md
CHANGED
@@ -53,11 +53,11 @@ from PIL import Image
|
|
53 |
from colpali_engine.models import ColQwen2, ColQwen2Processor
|
54 |
|
55 |
model = ColQwen2.from_pretrained(
|
56 |
-
"
|
57 |
torch_dtype=torch.bfloat16,
|
58 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|
59 |
)
|
60 |
-
processor = ColQwen2Processor.from_pretrained("
|
61 |
|
62 |
# Your inputs
|
63 |
images = [
|
|
|
53 |
from colpali_engine.models import ColQwen2, ColQwen2Processor
|
54 |
|
55 |
model = ColQwen2.from_pretrained(
|
56 |
+
"vidore/colqwen2-v0.1",
|
57 |
torch_dtype=torch.bfloat16,
|
58 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|
59 |
)
|
60 |
+
processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v0.1")
|
61 |
|
62 |
# Your inputs
|
63 |
images = [
|