Update README.md
Browse files
README.md
CHANGED
@@ -68,7 +68,7 @@ def main() -> None:
|
|
68 |
|
69 |
# Load model
|
70 |
model_name = "vidore/colpali"
|
71 |
-
model = ColPali.from_pretrained("
|
72 |
model.load_adapter(model_name)
|
73 |
processor = AutoProcessor.from_pretrained(model_name)
|
74 |
|
|
|
68 |
|
69 |
# Load model
|
70 |
model_name = "vidore/colpali"
|
71 |
+
model = ColPali.from_pretrained("vidore/colpaligemma-3b-mix-448-base", torch_dtype=torch.bfloat16, device_map="cuda").eval()
|
72 |
model.load_adapter(model_name)
|
73 |
processor = AutoProcessor.from_pretrained(model_name)
|
74 |
|