We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9aa5d73 commit 6bba8edCopy full SHA for 6bba8ed
include/llama.h
@@ -923,7 +923,7 @@ extern "C" {
923
924
// Get the embeddings for a sequence id
925
// Returns NULL if pooling_type is LLAMA_POOLING_TYPE_NONE
926
- // when pooling_type == LLAMA_POOLING_TYPE_RANK, returns float[i] with the rank(s) of the sequence
+ // when pooling_type == LLAMA_POOLING_TYPE_RANK, returns float[n_cls_out] with the rank(s) of the sequence
927
// otherwise: float[n_embd] (1-dimensional)
928
LLAMA_API float * llama_get_embeddings_seq(struct llama_context * ctx, llama_seq_id seq_id);
929
0 commit comments