| 232 | |
| 233 | |
| 234 | def save_to_config(ckpt_model_name, vae_model_name, hypernetwork_model_name, vram_usage_level): |
| 235 | config = getConfig() |
| 236 | if "model" not in config: |
| 237 | config["model"] = {} |
| 238 | |
| 239 | config["model"]["stable-diffusion"] = ckpt_model_name |
| 240 | config["model"]["vae"] = vae_model_name |
| 241 | config["model"]["hypernetwork"] = hypernetwork_model_name |
| 242 | |
| 243 | if vae_model_name is None or vae_model_name == "": |
| 244 | del config["model"]["vae"] |
| 245 | if hypernetwork_model_name is None or hypernetwork_model_name == "": |
| 246 | del config["model"]["hypernetwork"] |
| 247 | |
| 248 | config["vram_usage_level"] = vram_usage_level |
| 249 | |
| 250 | setConfig(config) |
| 251 | |
| 252 | |
| 253 | def update_render_threads(): |