Adding an Indexer class.

JuliaGenAI · May 30, 2024 · 0dc7cb8 · 0dc7cb8
1 parent 52d345e
commit 0dc7cb8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/ColBERT.jl b/src/ColBERT.jl
@@ -19,4 +19,8 @@ export RunSettings, TokenizerSettings, ResourceSettings,
 include("modelling/checkpoint.jl")
 export BaseColBERT, Checkpoint
 
+# indexer
+include("indexing.jl")
+export Indexer
+
 end
diff --git a/src/indexing.jl b/src/indexing.jl
@@ -0,0 +1,9 @@
+using .ColBERT: ColBERTConfig
+
+struct Indexer
+    # index_path::String            we can just reuse the path from the config?
+    checkpoint::String
+    config::ColBERTConfig
+end
+
+