oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-12-06 07:12:10 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Only save extensions settings on manual save

Browse source
This commit is contained in:
oobabooga 2025-06-15 15:53:16 -07:00
parent cc757f6226
commit bc2b0f54e9
1 changed files with 22 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
modules/ui.py
View file

@ -6,6 +6,7 @@ import gradio as gr
import yaml import yaml
import extensions import extensions
import modules.extensions as extensions_module
from modules import shared from modules import shared
from modules.chat import load_history from modules.chat import load_history
from modules.utils import gradio from modules.utils import gradio
@ -334,23 +335,11 @@ def save_settings(state, preset, extensions_list, show_controls, theme_state, ma
output.pop('instruction_template_str') output.pop('instruction_template_str')
output.pop('truncation_length') output.pop('truncation_length')
# Only save extensions on manual save # Handle extensions and extension parameters
if manual_save: if manual_save:
# Save current extensions and their parameter values
output['default_extensions'] = extensions_list output['default_extensions'] = extensions_list
else:
# Preserve existing extensions from settings file during autosave
settings_path = Path('user_data') / 'settings.yaml'
if settings_path.exists():
try:
with open(settings_path, 'r', encoding='utf-8') as f:
existing_settings = yaml.safe_load(f.read()) or {}
if 'default_extensions' in existing_settings:
output['default_extensions'] = existing_settings['default_extensions']
except Exception:
pass # If we can't read the file, just don't modify extensions
# Save extension values in the UI
for extension_name in extensions_list: for extension_name in extensions_list:
extension = getattr(extensions, extension_name, None) extension = getattr(extensions, extension_name, None)
if extension: if extension:
@ -362,6 +351,24 @@ def save_settings(state, preset, extensions_list, show_controls, theme_state, ma
# Only save if different from default value # Only save if different from default value
if param not in shared.default_settings or params[param] != shared.default_settings[param]: if param not in shared.default_settings or params[param] != shared.default_settings[param]:
output[_id] = params[param] output[_id] = params[param]
else:
# Preserve existing extensions and extension parameters during autosave
settings_path = Path('user_data') / 'settings.yaml'
if settings_path.exists():
try:
with open(settings_path, 'r', encoding='utf-8') as f:
existing_settings = yaml.safe_load(f.read()) or {}
# Preserve default_extensions
if 'default_extensions' in existing_settings:
output['default_extensions'] = existing_settings['default_extensions']
# Preserve extension parameter values
for key, value in existing_settings.items():
if any(key.startswith(f"{ext_name}-") for ext_name in extensions_module.available_extensions):
output[key] = value
except Exception:
pass # If we can't read the file, just don't modify extensions
# Do not save unchanged settings # Do not save unchanged settings
for key in list(output.keys()): for key in list(output.keys()):