@@ -116,7 +116,7 @@ struct common_sampler {
116116 const llama_model * model = llama_get_model (ctx);
117117 const llama_vocab * vocab = llama_model_get_vocab (model);
118118
119- const int n_vocab = llama_n_vocab (vocab);
119+ const int n_vocab = llama_vocab_n_vocab (vocab);
120120
121121 cur.resize (n_vocab);
122122
@@ -162,7 +162,7 @@ struct common_sampler * common_sampler_init(const struct llama_model * model, co
162162
163163 llama_sampler_chain_add (result->chain ,
164164 llama_sampler_init_logit_bias (
165- llama_n_vocab (vocab),
165+ llama_vocab_n_vocab (vocab),
166166 params.logit_bias .size (),
167167 params.logit_bias .data ()));
168168
@@ -177,7 +177,7 @@ struct common_sampler * common_sampler_init(const struct llama_model * model, co
177177 c_breakers.push_back (str.c_str ());
178178 }
179179
180- llama_sampler_chain_add (result->chain , llama_sampler_init_dry (vocab, llama_n_ctx_train (model), params.dry_multiplier , params.dry_base , params.dry_allowed_length , params.dry_penalty_last_n , c_breakers.data (), c_breakers.size ()));
180+ llama_sampler_chain_add (result->chain , llama_sampler_init_dry (vocab, llama_model_n_ctx_train (model), params.dry_multiplier , params.dry_base , params.dry_allowed_length , params.dry_penalty_last_n , c_breakers.data (), c_breakers.size ()));
181181 }
182182 break ;
183183 case COMMON_SAMPLER_TYPE_TOP_K:
@@ -211,7 +211,7 @@ struct common_sampler * common_sampler_init(const struct llama_model * model, co
211211 llama_sampler_chain_add (result->chain , llama_sampler_init_dist (params.seed ));
212212 } else if (params.mirostat == 1 ) {
213213 llama_sampler_chain_add (result->chain , llama_sampler_init_temp (params.temp ));
214- llama_sampler_chain_add (result->chain , llama_sampler_init_mirostat (llama_n_vocab (vocab), params.seed , params.mirostat_tau , params.mirostat_eta , 100 ));
214+ llama_sampler_chain_add (result->chain , llama_sampler_init_mirostat (llama_vocab_n_vocab (vocab), params.seed , params.mirostat_tau , params.mirostat_eta , 100 ));
215215 } else if (params.mirostat == 2 ) {
216216 llama_sampler_chain_add (result->chain , llama_sampler_init_temp (params.temp ));
217217 llama_sampler_chain_add (result->chain , llama_sampler_init_mirostat_v2 (params.seed , params.mirostat_tau , params.mirostat_eta ));
0 commit comments