Adding fields in the ResidualCodec type. Using @proto during dev.

JuliaGenAI · Jun 14, 2024 · beb65be · beb65be
1 parent e035aad
commit beb65be
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/indexing/codecs/residual.jl b/src/indexing/codecs/residual.jl
@@ -1,2 +1,10 @@
-struct ResidualCodec
+using ..ColBERT: ColBERTConfig 
+using ProtoStructs
+
+@proto mutable struct ResidualCodec
+    config::ColBERTConfig
+    centroids::Matrix{Float64}
+    avg_residual:: Float64
+    bucket_cutoffs::Vector{Float64}
+    bucket_weights::Vector{Float64}
 end