diff options
Diffstat (limited to 'examples/common.h')
| -rw-r--r-- | examples/common.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/examples/common.h b/examples/common.h index 6c2953cb..71332017 100644 --- a/examples/common.h +++ b/examples/common.h @@ -9,6 +9,7 @@ #include <random> #include <thread> #include <unordered_map> +#include <tuple> #if !defined (_WIN32) #include <stdio.h> @@ -95,7 +96,7 @@ std::vector<llama_token> llama_tokenize(struct llama_context * ctx, const std::s // Model utils // -struct llama_context * llama_init_from_gpt_params(const gpt_params & params); +std::tuple<struct llama_model *, struct llama_context *> llama_init_from_gpt_params(const gpt_params & params); // // Console utils |