Bläddra i källkod

embedding : enable --no-warmup option (#11475)

This commit enables the `--no-warmup` option for the llama-embeddings.

The motivation for this change is to allow the user to disable the
warmup when running the the program.
Daniel Bevenius 11 månader sedan
förälder
incheckning
b636228c0a
1 ändrade filer med 1 tillägg och 1 borttagningar
  1. 1 1
      common/arg.cpp

+ 1 - 1
common/arg.cpp

@@ -877,7 +877,7 @@ common_params_context common_params_parser_init(common_params & params, llama_ex
         [](common_params & params) {
         [](common_params & params) {
             params.warmup = false;
             params.warmup = false;
         }
         }
-    ).set_examples({LLAMA_EXAMPLE_MAIN, LLAMA_EXAMPLE_SERVER}));
+    ).set_examples({LLAMA_EXAMPLE_MAIN, LLAMA_EXAMPLE_SERVER, LLAMA_EXAMPLE_EMBEDDING}));
     add_opt(common_arg(
     add_opt(common_arg(
         {"--spm-infill"},
         {"--spm-infill"},
         string_format(
         string_format(