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