Adding some preliminary structs for storing models and checkpoints.

JuliaGenAI · May 24, 2024 · 962a947 · 962a947
1 parent 13507c6
commit 962a947
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/modelling/checkpoint.jl b/src/modelling/checkpoint.jl
@@ -0,0 +1,5 @@
+struct Checkpoint
+    model
+    doc_tokenizer
+    colbert_config
+end
diff --git a/src/modelling/colbert.jl b/src/modelling/colbert.jl
@@ -0,0 +1,5 @@
+struct ColBERT
+    bert
+    linear
+    tokenizer
+end