Update README.md
Browse files
README.md
CHANGED
@@ -71,6 +71,7 @@ def main() -> None:
|
|
71 |
model_name = "vidore/colpali-v1.1"
|
72 |
model = ColPali.from_pretrained("vidore/colpaligemma-3b-mix-448-base", torch_dtype=torch.bfloat16, device_map="cuda").eval()
|
73 |
model.load_adapter(model_name)
|
|
|
74 |
processor = AutoProcessor.from_pretrained(model_name)
|
75 |
|
76 |
# select images -> load_from_pdf(<pdf_path>), load_from_image_urls(["<url_1>"]), load_from_dataset(<path>)
|
|
|
71 |
model_name = "vidore/colpali-v1.1"
|
72 |
model = ColPali.from_pretrained("vidore/colpaligemma-3b-mix-448-base", torch_dtype=torch.bfloat16, device_map="cuda").eval()
|
73 |
model.load_adapter(model_name)
|
74 |
+
model = model.eval()
|
75 |
processor = AutoProcessor.from_pretrained(model_name)
|
76 |
|
77 |
# select images -> load_from_pdf(<pdf_path>), load_from_image_urls(["<url_1>"]), load_from_dataset(<path>)
|