summaryrefslogtreecommitdiff
path: root/llama.cpp
diff options
context:
space:
mode:
authorCebtenzzre <cebtenzzre@gmail.com>2023-09-08 11:43:35 -0400
committerGitHub <noreply@github.com>2023-09-08 11:43:35 -0400
commite64f5b55783e910d8287363895d652b4bea6527a (patch)
treedba460e07999ff2d9d0c4a7499e5cf3c591f9748 /llama.cpp
parent94f10b91ed69980f299441e49c8dbdb448f0ccc6 (diff)
examples : make n_ctx warning work again (#3066)
This was broken by commit e36ecdcc ("build : on Mac OS enable Metal by default (#2901)").
Diffstat (limited to 'llama.cpp')
-rw-r--r--llama.cpp14
1 files changed, 11 insertions, 3 deletions
diff --git a/llama.cpp b/llama.cpp
index 3f119022..2a2a0c9c 100644
--- a/llama.cpp
+++ b/llama.cpp
@@ -5633,15 +5633,19 @@ void llama_free(struct llama_context * ctx) {
}
int llama_n_vocab(const struct llama_context * ctx) {
- return ctx->model.vocab.id_to_token.size();
+ return llama_model_n_vocab(&ctx->model);
}
int llama_n_ctx(const struct llama_context * ctx) {
- return ctx->model.hparams.n_ctx;
+ return llama_model_n_ctx(&ctx->model);
+}
+
+int llama_n_ctx_train(const struct llama_context * ctx) {
+ return llama_model_n_ctx_train(&ctx->model);
}
int llama_n_embd(const struct llama_context * ctx) {
- return ctx->model.hparams.n_embd;
+ return llama_model_n_embd(&ctx->model);
}
enum llama_vocab_type llama_vocab_type(const struct llama_context * ctx) {
@@ -5656,6 +5660,10 @@ int llama_model_n_ctx(const struct llama_model * model) {
return model->hparams.n_ctx;
}
+int llama_model_n_ctx_train(const struct llama_model * model) {
+ return model->hparams.n_ctx_train;
+}
+
int llama_model_n_embd(const struct llama_model * model) {
return model->hparams.n_embd;
}