diff options
| author | Douglas Hanley <thesecretaryofwar@gmail.com> | 2024-04-24 08:10:07 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-24 16:10:07 +0300 |
| commit | b4e4b8a9351d918a56831c73cf9f25c1837b80d1 (patch) | |
| tree | 26aa8b15ca29b3e63ab300cc863739f1f31cd1d6 /common/common.h | |
| parent | 3fe847b5747676ee1bf90371c46ed0bc66b57240 (diff) | |
llama : add llama_get_pooling_type function (#6862)
* add llama_get_pooling_type function
* fix argument name, move with ctx funcs
Diffstat (limited to 'common/common.h')
| -rw-r--r-- | common/common.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/common/common.h b/common/common.h index 157b54a3..87361e8e 100644 --- a/common/common.h +++ b/common/common.h @@ -86,8 +86,8 @@ struct gpt_params { ggml_numa_strategy numa = GGML_NUMA_STRATEGY_DISABLED; - llama_rope_scaling_type rope_scaling_type = LLAMA_ROPE_SCALING_TYPE_UNSPECIFIED; - llama_pooling_type pooling_type = LLAMA_POOLING_TYPE_UNSPECIFIED; // pooling type for embeddings + enum llama_rope_scaling_type rope_scaling_type = LLAMA_ROPE_SCALING_TYPE_UNSPECIFIED; + enum llama_pooling_type pooling_type = LLAMA_POOLING_TYPE_UNSPECIFIED; // pooling type for embeddings // // sampling parameters struct llama_sampling_params sparams; |