AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

When saving a preset, only save params that differ from the defaults

Browse Source
This commit is contained in:
oobabooga 2023-07-31 19:13:29 -07:00
parent 84297d05c4
commit f094330df0
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
modules/presets.py
View File

@ -4,8 +4,8 @@ from pathlib import Path
import yaml
def load_preset(name):
generate_params = {
def default_preset():
return {
'do_sample': True,
'temperature': 1,
'top_p': 1,
@ -29,6 +29,9 @@ def load_preset(name):
'mirostat_eta': 0.1,
}
def load_preset(name):
generate_params = default_preset()
if name not in ['None', None, '']:
with open(Path(f'presets/{name}.yaml'), 'r') as infile:
preset = yaml.safe_load(infile)
@ -52,5 +55,12 @@ def load_preset_for_ui(name, state):
def generate_preset_yaml(state):
defaults = default_preset()
data = {k: state[k] for k in ['do_sample', 'temperature', 'top_p', 'typical_p', 'epsilon_cutoff', 'eta_cutoff', 'repetition_penalty', 'repetition_penalty_range', 'encoder_repetition_penalty', 'top_k', 'min_length', 'no_repeat_ngram_size', 'num_beams', 'penalty_alpha', 'length_penalty', 'early_stopping', 'mirostat_mode', 'mirostat_tau', 'mirostat_eta', 'tfs', 'top_a']}
# Remove entries that are identical to the defaults
for k in list(data.keys()):
if data[k] == defaults[k]:
del data[k]
return yaml.dump(data, sort_keys=False)