ColPali
Safetensors
English
vidore
manu commited on
Commit
a57445e
1 Parent(s): 6b26582

Update README.md

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