Update README.md
Browse files
README.md
CHANGED
@@ -63,7 +63,7 @@ def main() -> None:
|
|
63 |
"""Example script to run inference with ColPali"""
|
64 |
|
65 |
# Load model
|
66 |
-
model_name = "vidore/colpali-v1.
|
67 |
model = ColPali.from_pretrained("vidore/colpaligemma-3b-pt-448-base", torch_dtype=torch.bfloat16, device_map="cuda").eval()
|
68 |
model.load_adapter(model_name)
|
69 |
processor = AutoProcessor.from_pretrained(model_name)
|
|
|
63 |
"""Example script to run inference with ColPali"""
|
64 |
|
65 |
# Load model
|
66 |
+
model_name = "vidore/colpali-v1.2"
|
67 |
model = ColPali.from_pretrained("vidore/colpaligemma-3b-pt-448-base", torch_dtype=torch.bfloat16, device_map="cuda").eval()
|
68 |
model.load_adapter(model_name)
|
69 |
processor = AutoProcessor.from_pretrained(model_name)
|