Function setConfig

ui/easydiffusion/app.py:191–231

(config)

def setConfig(config):
    global MODELS_DIR

    try:  # config.yaml
        config_yaml_path = os.path.join(CONFIG_DIR, "..", "config.yaml")
        config_yaml_path = os.path.abspath(config_yaml_path)
        yaml = YAML()

        if not hasattr(config, "_yaml_comment"):
            config_yaml_sample_path = os.path.join(CONFIG_DIR, "config.yaml.sample")

            if os.path.exists(config_yaml_sample_path):
                with open(config_yaml_sample_path, "r", encoding="utf-8") as f:
                    commented_config = yaml.load(f)

                for k in config:
                    commented_config[k] = config[k]

                config = commented_config
        yaml.indent(mapping=2, sequence=4, offset=2)

        if "config_on_startup" in config:
            del config["config_on_startup"]

        try:
            f = open(config_yaml_path + ".tmp", "w", encoding="utf-8")
            yaml.dump(config, f)
        finally:
            f.close()  # do this explicitly to avoid NUL bytes (possible rare bug when using 'with')

        # verify that the new file is valid, and only then overwrite the old config file
        # helps prevent the rare NUL bytes error from corrupting the config file
        yaml = YAML()
        with open(config_yaml_path + ".tmp", "r", encoding="utf-8") as f:
            yaml.load(f)
        shutil.move(config_yaml_path + ".tmp", config_yaml_path)
    except:
        log.error(traceback.format_exc())

    if config.get("models_dir"):
        MODELS_DIR = config["models_dir"]


def save_to_config(ckpt_model_name, vae_model_name, hypernetwork_model_name, vram_usage_level):