vidore
/

colpali-v1.2-merged

vidore-experimental

Model card Files Files and versions Community

tonywu71 commited on Oct 7, 2024

Commit

496d826

·

verified ·

1 Parent(s): 541c3fe

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -73,7 +73,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("google/paligemma-3b-mix-448"))

         "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("google/paligemma-3b-mix-448"))