Allow CFG on impersonation

This commit is contained in:
Cohee
2023-08-22 19:35:56 +03:00
parent 38a08cd887
commit 6923044a34

View File

@@ -168,9 +168,9 @@ $(document).ready(function () {
textgenerationwebui_settings[id] = value; textgenerationwebui_settings[id] = value;
} }
else { else {
const value = parseFloat($(this).val()); const value = Number($(this).val());
$(`#${id}_counter_textgenerationwebui`).text(value.toFixed(2)); $(`#${id}_counter_textgenerationwebui`).text(value.toFixed(2));
textgenerationwebui_settings[id] = parseFloat(value); textgenerationwebui_settings[id] = value;
} }
saveSettingsDebounced(); saveSettingsDebounced();
@@ -207,7 +207,7 @@ async function generateTextGenWithStreaming(generate_data, signal) {
const response = await fetch('/generate_textgenerationwebui', { const response = await fetch('/generate_textgenerationwebui', {
headers: { headers: {
...getRequestHeaders(), ...getRequestHeaders(),
'X-Response-Streaming': true, 'X-Response-Streaming': String(true),
'X-Streaming-URL': textgenerationwebui_settings.streaming_url, 'X-Streaming-URL': textgenerationwebui_settings.streaming_url,
}, },
body: JSON.stringify(generate_data), body: JSON.stringify(generate_data),
@@ -251,8 +251,8 @@ export function getTextGenGenerationData(finalPromt, this_amount_gen, isImperson
'penalty_alpha': textgenerationwebui_settings.penalty_alpha, 'penalty_alpha': textgenerationwebui_settings.penalty_alpha,
'length_penalty': textgenerationwebui_settings.length_penalty, 'length_penalty': textgenerationwebui_settings.length_penalty,
'early_stopping': textgenerationwebui_settings.early_stopping, 'early_stopping': textgenerationwebui_settings.early_stopping,
'guidance_scale': isImpersonate ? 1 : cfgValues?.guidanceScale?.value ?? textgenerationwebui_settings.guidance_scale ?? 1, 'guidance_scale': cfgValues?.guidanceScale?.value ?? textgenerationwebui_settings.guidance_scale ?? 1,
'negative_prompt': isImpersonate ? '' : cfgValues?.negativePrompt ?? textgenerationwebui_settings.negative_prompt ?? '', 'negative_prompt': cfgValues?.negativePrompt ?? textgenerationwebui_settings.negative_prompt ?? '',
'seed': textgenerationwebui_settings.seed, 'seed': textgenerationwebui_settings.seed,
'add_bos_token': textgenerationwebui_settings.add_bos_token, 'add_bos_token': textgenerationwebui_settings.add_bos_token,
'stopping_strings': getStoppingStrings(isImpersonate, false), 'stopping_strings': getStoppingStrings(isImpersonate, false),