diff options
| -rw-r--r-- | examples/common.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/examples/common.cpp b/examples/common.cpp index 9b23b1f..222b4fa 100644 --- a/examples/common.cpp +++ b/examples/common.cpp @@ -414,6 +414,8 @@ struct llama_context * llama_init_from_gpt_params(const gpt_params & params) { lparams.f16_kv = params.memory_f16; lparams.use_mmap = params.use_mmap; lparams.use_mlock = params.use_mlock; + lparams.logits_all = params.perplexity; + lparams.embedding = params.embedding; llama_context * lctx = llama_init_from_file(params.model.c_str(), lparams); |