add max_tokens_second

This commit is contained in:
gabriel dhimoila
2024-02-29 00:55:25 +01:00
parent 6f5dbc2a52
commit 76669ff8bb
3 changed files with 10 additions and 1 deletions

View File

@ -111,6 +111,7 @@ const settings = {
max_temp: 2.0,
dynatemp_exponent: 1.0,
smoothing_factor: 0.0,
max_tokens_second: 0,
seed: -1,
preset: 'Default',
add_bos_token: true,
@ -178,6 +179,7 @@ const setting_names = [
'max_temp',
'dynatemp_exponent',
'smoothing_factor',
'max_tokens_second',
'encoder_rep_pen',
'freq_pen',
'presence_pen',
@ -984,6 +986,7 @@ export function getTextGenGenerationData(finalPrompt, maxTokens, isImpersonate,
'dynatemp_range': settings.dynatemp ? (settings.max_temp - settings.min_temp) / 2 : 0,
'dynatemp_exponent': settings.dynatemp ? settings.dynatemp_exponent : 1,
'smoothing_factor': settings.smoothing_factor,
'max_tokens_second': settings.max_tokens_second,
'sampler_priority': settings.type === OOBA ? settings.sampler_priority : undefined,
'samplers': settings.type === LLAMACPP ? settings.samplers : undefined,
'stopping_strings': getStoppingStrings(isImpersonate, isContinue),