From e035aada3b02e26bd2426548d65ec547332d830d Mon Sep 17 00:00:00 2001 From: Siddhant Chaudhary Date: Fri, 14 Jun 2024 22:05:18 +0530 Subject: [PATCH] Explicitly casting to `Float64`s. --- src/indexing/collection_encoder.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/indexing/collection_encoder.jl b/src/indexing/collection_encoder.jl index 9615f75..733bebf 100644 --- a/src/indexing/collection_encoder.jl +++ b/src/indexing/collection_encoder.jl @@ -12,7 +12,7 @@ function encode_passages(encoder::CollectionEncoder, passages::Vector{String}) error("The list of passages to encode is empty!") end - embs, doclens = Vector{Matrix}(), Vector{Int}() + embs, doclens = Vector{Matrix{Float64}}(), Vector{Int}() # batching here to avoid storing intermediate embeddings on GPU # batching also occurs inside docFromText to do batch packing optimizations for passages_batch in batch(passages, encoder.config.indexing_settings.index_bsize * 50)