Skip to content

Commit

Permalink
Explicitly casting to Float64s.
Browse files Browse the repository at this point in the history
  • Loading branch information
codetalker7 committed Jun 14, 2024
1 parent 2c9d80f commit e035aad
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/indexing/collection_encoder.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit e035aad

Please sign in to comment.