Skip to content

Commit 54cf43a

Browse files
committed
rnn fix adjust
1 parent 5d35193 commit 54cf43a

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

gpttype_adapter.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)