File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -4452,20 +4452,20 @@ generation_outputs gpttype_generate(const generation_inputs inputs)
44524452
44534453 // if running rnn model in smartcache mode, save progress a little bit before the final PP is done
44544454 // this helps solve token boundary mutation issues
4455- if (draft_ctx==nullptr && embd.size ()>1 && !startedsampling && input_consumed==embd_inp.size () && input_consumed>128 )
4455+ if (draft_ctx==nullptr && embd.size ()>1 && !startedsampling && input_consumed==embd_inp.size () && input_consumed>64 )
44564456 {
44574457 if (kcpp_data->smartcache && is_recurrent && file_format==FileFormat::GGUF_GENERIC && current_context_tokens.size () > 32 )
44584458 {
4459- if (embd.size ()<=64 )
4459+ if (embd.size ()<=48 )
44604460 {
44614461 // directly snapshot for a small batch
44624462 smartcache_quick_snapshot ();
44634463 }
44644464 else
44654465 {
44664466 skipdecodelater = true ;
4467- // decode until nearly done, then snapshot and decode the last 64
4468- std::vector<std::vector<gpt_vocab::id>> parts = split_big_vector_in_two (embd,64 );
4467+ // decode until nearly done, then snapshot and decode the last 32
4468+ std::vector<std::vector<gpt_vocab::id>> parts = split_big_vector_in_two (embd,32 );
44694469 int temp_past = n_past;
44704470 evalres = true ;
44714471 for (int p=0 ;p<parts.size ();++p)
You can’t perform that action at this time.
0 commit comments