diff options
| author | Georgi Gerganov <ggerganov@gmail.com> | 2023-07-01 18:46:00 +0300 |
|---|---|---|
| committer | Georgi Gerganov <ggerganov@gmail.com> | 2023-07-01 18:46:00 +0300 |
| commit | 79f634a19d1c32a6cfb1befc21551ee684fced6b (patch) | |
| tree | be75c79f53ea2c357f39aeda132c9272baf669bd /examples/embd-input/embd-input.h | |
| parent | 04606a159947566b27810508433e6ca5dbc684ba (diff) | |
embd-input : fix returning ptr to temporary
Diffstat (limited to 'examples/embd-input/embd-input.h')
| -rw-r--r-- | examples/embd-input/embd-input.h | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/examples/embd-input/embd-input.h b/examples/embd-input/embd-input.h index 4fefabd4..efb5ba5e 100644 --- a/examples/embd-input/embd-input.h +++ b/examples/embd-input/embd-input.h @@ -5,7 +5,6 @@ #include "llama.h" #include "build-info.h" - extern "C" { typedef struct MyModel { @@ -14,14 +13,13 @@ typedef struct MyModel { int n_past = 0; } MyModel; - struct MyModel* create_mymodel(int argc, char ** argv); bool eval_float(void* model, float* input, int N); bool eval_tokens(void* model, std::vector<llama_token> tokens); bool eval_id(struct MyModel* mymodel, int id); bool eval_string(struct MyModel* mymodel, const char* str); -const char* sampling(struct MyModel* mymodel); +const char * sampling(struct MyModel* mymodel); llama_token sampling_id(struct MyModel* mymodel); void free_mymodel(struct MyModel* mymodel); |