vidore
/

colpali-v1.1

Visual Document Retrieval

Model card Files Files and versions Community

manu commited on Aug 21, 2024

Commit

8949357

·

verified ·

1 Parent(s): b434f0e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -61,7 +61,7 @@ def main() -> None:
     """Example script to run inference with ColPali"""
     # Load model
-    model_name = "manu/colpali-v1.1"
     model = ColPali.from_pretrained("vidore/colpaligemma-3b-mix-448-base", torch_dtype=torch.bfloat16, device_map="cuda").eval()
     model.load_adapter(model_name)
     processor = AutoProcessor.from_pretrained(model_name)

     """Example script to run inference with ColPali"""
     # Load model
+    model_name = "vidore/colpali-v1.1"
     model = ColPali.from_pretrained("vidore/colpaligemma-3b-mix-448-base", torch_dtype=torch.bfloat16, device_map="cuda").eval()
     model.load_adapter(model_name)
     processor = AutoProcessor.from_pretrained(model_name)