Skip to content

Commit 2b18051

Browse files
authored
minor : add TODO [no ci]
1 parent 11c17d7 commit 2b18051

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/llama-model.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19860,6 +19860,7 @@ ggml_cgraph * llama_model::build_graph(const llm_graph_params & params) const {
1986019860
// if the gguf model was converted with --sentence-transformers-dense-modules
1986119861
// there will be two additional dense projection layers
1986219862
// dense linear projections are applied after pooling
19863+
// TODO: move reranking logic here and generalize
1986319864
llm->build_dense_out(dense_2_out_layers, dense_3_out_layers);
1986419865

1986519866
return llm->res->get_gf();

0 commit comments

Comments
 (0)