summaryrefslogtreecommitdiff
path: root/examples/lookup/lookup.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'examples/lookup/lookup.cpp')
-rw-r--r--examples/lookup/lookup.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/lookup/lookup.cpp b/examples/lookup/lookup.cpp
index d53a9828..1fff4f74 100644
--- a/examples/lookup/lookup.cpp
+++ b/examples/lookup/lookup.cpp
@@ -106,7 +106,7 @@ int main(int argc, char ** argv){
bool has_eos = false;
- struct llama_sampling_context * ctx_sampling = llama_sampling_init(params.sparams);
+ struct llama_sampling_context * ctx_sampling = llama_sampling_init(llama_get_model_vocab(model), params.sparams);
std::vector<llama_token> draft;