vidore
/

colpali-v1.2

vidore-experimental

Model card Files Files and versions Community

tonywu71 commited on Oct 7

Commit

fa46f8e

•

1 Parent(s): 9912ce6

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -76,7 +76,7 @@ model = cast(
         "vidore/colpali-v1.2",
         torch_dtype=torch.bfloat16,
         device_map="cuda:0",  # or "mps" if on Apple Silicon
-    ),
 )
 processor = cast(ColPaliProcessor, ColPaliProcessor.from_pretrained("vidore/colpali-v1.2"))

         "vidore/colpali-v1.2",
         torch_dtype=torch.bfloat16,
         device_map="cuda:0",  # or "mps" if on Apple Silicon
+    ).eval()
 )
 processor = cast(ColPaliProcessor, ColPaliProcessor.from_pretrained("vidore/colpali-v1.2"))