mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

UI: Do not save unchanged extension settings to settings.yaml

Browse source
This commit is contained in:
oobabooga 2024-01-10 03:48:30 -08:00
parent bb2c4707c4
commit b3fc2cd887
2 changed files with 13 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

17
modules/extensions.py
View file

@ -19,10 +19,9 @@ def apply_settings(extension, name):
for param in extension.params:
_id = f"{name}-{param}"
if _id not in shared.settings:
continue
extension.params[param] = shared.settings[_id]
shared.default_settings[_id] = extension.params[param]
if _id in shared.settings:
extension.params[param] = shared.settings[_id]
def load_extensions():
@ -40,10 +39,14 @@ def load_extensions():
raise
extension = getattr(extensions, name).script
apply_settings(extension, name)
if extension not in setup_called and hasattr(extension, "setup"):
# Only run setup() and apply settings from settings.yaml once
if extension not in setup_called:
apply_settings(extension, name)
if hasattr(extension, "setup"):
extension.setup()
setup_called.add(extension)
extension.setup()
state[name] = [True, i]
except: