Updated example [skip ci]

Author: ankane

examples/colpali/exact.py

@@ -37,11 +37,11 @@
 
 def generate_embeddings(processed):
     with torch.no_grad():
-        return model(**processed.to(model.device))
+        return model(**processed.to(model.device)).to(torch.float32).numpy(force=True)
 
 
 def binary_quantize(embedding):
-    return Bit(embedding.gt(0).numpy(force=True))
+    return Bit(embedding > 0)
 
 
 input = load_dataset('vidore/docvqa_test_subsampled', split='test[:3]')['image']