diff options
Diffstat (limited to 'examples/infill/infill.cpp')
| -rw-r--r-- | examples/infill/infill.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/infill/infill.cpp b/examples/infill/infill.cpp index 92d630b1..d3c3ad5a 100644 --- a/examples/infill/infill.cpp +++ b/examples/infill/infill.cpp @@ -349,7 +349,7 @@ int main(int argc, char ** argv) { std::vector<llama_token> embd; - struct llama_sampling_context * ctx_sampling = llama_sampling_init(sparams); + struct llama_sampling_context * ctx_sampling = llama_sampling_init(llama_get_model_vocab(model), sparams); while (n_remain != 0 || params.interactive) { // predict |