From 67c77799e025a8425c23a6a0599c007f46ded590 Mon Sep 17 00:00:00 2001 From: Ron Evans Date: Tue, 2 May 2023 22:39:51 +0200 Subject: examples : add llama_init_from_gpt_params() common function (#1290) Signed-off-by: deadprogram --- examples/embedding/embedding.cpp | 22 ++++------------------ 1 file changed, 4 insertions(+), 18 deletions(-) (limited to 'examples/embedding') diff --git a/examples/embedding/embedding.cpp b/examples/embedding/embedding.cpp index 1e9d8a8..e4b7291 100644 --- a/examples/embedding/embedding.cpp +++ b/examples/embedding/embedding.cpp @@ -35,24 +35,10 @@ int main(int argc, char ** argv) { llama_context * ctx; // load the model - { - auto lparams = llama_context_default_params(); - - lparams.n_ctx = params.n_ctx; - lparams.n_parts = params.n_parts; - lparams.seed = params.seed; - lparams.f16_kv = params.memory_f16; - lparams.logits_all = params.perplexity; - lparams.use_mmap = params.use_mmap; - lparams.use_mlock = params.use_mlock; - lparams.embedding = params.embedding; - - ctx = llama_init_from_file(params.model.c_str(), lparams); - - if (ctx == NULL) { - fprintf(stderr, "%s: error: failed to load model '%s'\n", __func__, params.model.c_str()); - return 1; - } + ctx = llama_init_from_gpt_params(params); + if (ctx == NULL) { + fprintf(stderr, "%s: error: unable to load model\n", __func__); + return 1; } // print system information -- cgit v1.2.3