From beb65be776f761edd93b2ae4473e3fb69db72e01 Mon Sep 17 00:00:00 2001 From: Siddhant Chaudhary Date: Fri, 14 Jun 2024 22:07:05 +0530 Subject: [PATCH] Adding fields in the `ResidualCodec` type. Using `@proto` during dev. --- src/indexing/codecs/residual.jl | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/indexing/codecs/residual.jl b/src/indexing/codecs/residual.jl index 7697fef..e1c4421 100644 --- 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