extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-02 20:16:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix for endlessly nesting config.yaml

Browse Source
This commit is contained in:
cmdr2 2023-07-22 11:47:48 +05:30
parent 9fd69b2519
commit ade0912ba1
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ui/easydiffusion/app.py
View File

@ -102,6 +102,7 @@ def init_render_threads():
update_render_threads() update_render_threads()
def getConfig(default_val=APP_CONFIG_DEFAULTS): def getConfig(default_val=APP_CONFIG_DEFAULTS):
config_yaml_path = os.path.join(CONFIG_DIR, "..", "config.yaml") config_yaml_path = os.path.join(CONFIG_DIR, "..", "config.yaml")
@ -111,9 +112,9 @@ def getConfig(default_val=APP_CONFIG_DEFAULTS):
shutil.move(config_legacy_yaml, config_yaml_path) shutil.move(config_legacy_yaml, config_yaml_path)
def set_config_on_startup(config: dict): def set_config_on_startup(config: dict):
if (getConfig.__config_on_startup is None): if getConfig.__test_diffusers_on_startup is None:
getConfig.__config_on_startup = copy.deepcopy(config) getConfig.__test_diffusers_on_startup = config.get("test_diffusers", False)
config["config_on_startup"] = getConfig.__config_on_startup config["config_on_startup"] = {"test_diffusers": getConfig.__test_diffusers_on_startup}
if os.path.isfile(config_yaml_path): if os.path.isfile(config_yaml_path):
try: try:
@ -160,7 +161,8 @@ def getConfig(default_val=APP_CONFIG_DEFAULTS):
set_config_on_startup(default_val) set_config_on_startup(default_val)
return default_val return default_val
getConfig.__config_on_startup = None
getConfig.__test_diffusers_on_startup = None
def setConfig(config): def setConfig(config):
@ -181,6 +183,8 @@ def setConfig(config):
config = commented_config config = commented_config
yaml.indent(mapping=2, sequence=4, offset=2) yaml.indent(mapping=2, sequence=4, offset=2)
del config["config_on_startup"]
try: try:
f = open(config_yaml_path + ".tmp", "w", encoding="utf-8") f = open(config_yaml_path + ".tmp", "w", encoding="utf-8")
yaml.dump(config, f) yaml.dump(config, f)