summaryrefslogtreecommitdiff
path: root/examples/lookup/lookup-stats.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'examples/lookup/lookup-stats.cpp')
-rw-r--r--examples/lookup/lookup-stats.cpp3
1 files changed, 1 insertions, 2 deletions
diff --git a/examples/lookup/lookup-stats.cpp b/examples/lookup/lookup-stats.cpp
index 0b171c87..2fe67100 100644
--- a/examples/lookup/lookup-stats.cpp
+++ b/examples/lookup/lookup-stats.cpp
@@ -31,7 +31,6 @@ int main(int argc, char ** argv){
// load the model
std::tie(model, ctx) = llama_init_from_gpt_params(params);
- GGML_ASSERT(llama_n_vocab(model) < (1 << 16));
// tokenize the prompt
std::vector<llama_token> inp;
@@ -65,7 +64,7 @@ int main(int argc, char ** argv){
}
const int n_input = inp.size();
- const int n_ctx = params.n_ctx;
+ const int n_ctx = llama_n_ctx(ctx);
int n_drafted = 0;
int n_accept = 0;