Skip to content

Commit 815d205

Browse files
committed
gentoken reservations
1 parent 0ac20e3 commit 815d205

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

gpttype_adapter.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3603,6 +3603,9 @@ generation_outputs gpttype_generate(const generation_inputs inputs)
36033603
int input_consumed = 0;
36043604
std::mt19937 rng(kcpp_data->seed);
36053605

3606+
//do some reservation so we don't have to realloc
3607+
generated_tokens.reserve(remaining_tokens+16);
3608+
36063609
//prepare sampler order
36073610
std::vector<samplers> sampler_order;
36083611
if(inputs.sampler_len<=0) //list by value

0 commit comments

Comments
 (0)