Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
ggml.c : add gguf_find_key
  • Loading branch information
klosax authored Jul 28, 2023
commit 765fab636a31a12e2d053dab45e8651f1799966e
15 changes: 15 additions & 0 deletions ggml.c
Original file line number Diff line number Diff line change
Expand Up @@ -18745,6 +18745,21 @@ int gguf_get_n_kv(struct gguf_context * ctx) {
return ctx->header.n_kv;
}

int gguf_find_key(struct gguf_context * ctx, const char * key) {
// return -1 if key not found
const int n_kv = gguf_get_n_kv(ctx);
int keyfound = -1;

for (int i = 0; i < n_kv; ++i) {
if (strcmp(key, gguf_get_key(ctx, i)) == 0) {
keyfound = i;
break;
}
}

return keyfound;
}

const char * gguf_get_key(struct gguf_context * ctx, int i) {
return ctx->header.kv[i].key.data;
}
Expand Down