Smoothing curve support for ooba

This commit is contained in:
kalomaze
2024-03-01 00:06:34 -06:00
parent 2ef41ed883
commit 45776de1d5
3 changed files with 22 additions and 5 deletions

View File

@@ -111,6 +111,7 @@ const settings = {
max_temp: 2.0,
dynatemp_exponent: 1.0,
smoothing_factor: 0.0,
smoothing_curve: 1.0,
max_tokens_second: 0,
seed: -1,
preset: 'Default',
@@ -179,6 +180,7 @@ const setting_names = [
'max_temp',
'dynatemp_exponent',
'smoothing_factor',
'smoothing_curve',
'max_tokens_second',
'encoder_rep_pen',
'freq_pen',
@@ -647,6 +649,7 @@ jQuery(function () {
'typical_p_textgenerationwebui': 1, // Added entry
'guidance_scale_textgenerationwebui': 1,
'smoothing_factor_textgenerationwebui': 0,
'smoothing_curve_textgenerationwebui': 1,
};
for (const [id, value] of Object.entries(inputs)) {
@@ -986,6 +989,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,
'smoothing_curve': settings.smoothing_curve,
'max_tokens_second': settings.max_tokens_second,
'sampler_priority': settings.type === OOBA ? settings.sampler_priority : undefined,
'samplers': settings.type === LLAMACPP ? settings.samplers : undefined,