rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

#583 Add mirostat generation options for oobabooga + preset

Browse Source
This commit is contained in:
Cohee
2023-07-14 19:51:05 +03:00
parent 2ac77b21ae
commit 8f96a74d1e
32 changed files with 198 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
public/scripts/textgen-settings.js
View File

@@ -38,6 +38,9 @@ let textgenerationwebui_settings = {
skip_special_tokens: true,
streaming: false,
streaming_url: 'ws://127.0.0.1:5005/api/v1/stream',
mirostat_mode: 0,
mirostat_tau: 5,
mirostat_eta: 0.1,
};
let textgenerationwebui_presets = [];
@@ -67,6 +70,9 @@ const setting_names = [
"skip_special_tokens",
"streaming",
"streaming_url",
"mirostat_mode",
"mirostat_tau",
"mirostat_eta",
];
function selectPreset(name) {
@@ -224,5 +230,8 @@ export function getTextGenGenerationData(finalPromt, this_amount_gen, isImperson
'tfs': textgenerationwebui_settings.tfs,
'epsilon_cutoff': textgenerationwebui_settings.epsilon_cutoff,
'eta_cutoff': textgenerationwebui_settings.eta_cutoff,
'mirostat_mode': textgenerationwebui_settings.mirostat_mode,
'mirostat_tau': textgenerationwebui_settings.mirostat_tau,
'mirostat_eta': textgenerationwebui_settings.mirostat_eta,
};
}