Address review comments and add example notebook

TileDB-Inc · Oct 14, 2024 · b0cfff7 · b0cfff7
1 parent 77ffa26
commit b0cfff7
Show file tree

Hide file tree

Showing 3 changed files with 282 additions and 0 deletions.
diff --git a/apis/python/examples/object_api/multi_modal_pdf_search.ipynb b/apis/python/examples/object_api/multi_modal_pdf_search.ipynb
diff --git a/apis/python/src/tiledb/vector_search/embeddings/colpali_embedding.py b/apis/python/src/tiledb/vector_search/embeddings/colpali_embedding.py
@@ -41,6 +41,8 @@ def load(self) -> None:
         if self.device is None:
             if torch.cuda.is_available() and torch.cuda.device_count() > 0:
                 self.device = "cuda"
+            elif torch.backends.mps.is_available():
+                self.device = "mps"
             else:
                 self.device = "cpu"
 

diff --git a/apis/python/src/tiledb/vector_search/object_api/object_index.py b/apis/python/src/tiledb/vector_search/object_api/object_index.py
@@ -1020,6 +1020,7 @@ def create(
             schema = tiledb.ArraySchema(
                 domain=external_ids_dom,
                 sparse=True,
+                capacity=metadata_tile_size,
                 attrs=object_reader.metadata_attributes(),
             )
             tiledb.Array.create(object_metadata_array_uri, schema)