]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
model-conversion : set pooling type to none in logits.cpp (#15564)
authorDaniel Bevenius <redacted>
Mon, 25 Aug 2025 13:00:43 +0000 (15:00 +0200)
committerGitHub <redacted>
Mon, 25 Aug 2025 13:00:43 +0000 (15:00 +0200)
This commit explicitly sets the pooling type to 'none' in the logits.cpp
to support models that have a pooling type specified.

The motivation for this is that some models may have a pooling type set
in the model file (.gguf file) and for this specific case where we only
want to extract logits, we need to ensure that no pooling is used to
so that we are comparing raw logits and not pooled embeddings.

examples/model-conversion/logits.cpp

index 2cac6a3b3eaae90ccd565a652eaadb1a4f0b7644..ddc5e9005f9e00dcd25b87c7d645eae63d9bd203 100644 (file)
@@ -112,6 +112,7 @@ int main(int argc, char ** argv) {
     ctx_params.no_perf = false;
     if (embedding_mode) {
         ctx_params.embeddings = true;
+        ctx_params.pooling_type = LLAMA_POOLING_TYPE_NONE;
         ctx_params.n_ubatch = ctx_params.n_batch;
     }