ColPali
Safetensors
English
vidore
vidore-experimental
manu commited on
Commit
16716eb
1 Parent(s): 6629d45

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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>)