common : warm-up with 2 tokens - seems to work better

This commit is contained in:
Georgi Gerganov 2023-09-03 21:07:01 +03:00
parent 013457885a
commit ebe41d49a6
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -770,7 +770,7 @@ std::tuple<struct llama_model *, struct llama_context *> llama_init_from_gpt_par
{
LOG("warming up the model with an empty run\n");
const std::vector<llama_token> tmp = { llama_token_bos(lctx), };
const std::vector<llama_token> tmp = { llama_token_bos(lctx), llama_token_eos(lctx), };
llama_eval(lctx, tmp.data(), tmp.size(), 0, params.n_threads);
llama_reset_timings(lctx);
}