From 86e3ccaa9d866e8e53e41db32f5e39cca34eff40 Mon Sep 17 00:00:00 2001 From: altoiddealer Date: Fri, 23 Aug 2024 12:01:30 -0400 Subject: [PATCH] checkpoint change (#1427) * checkpoint change credit to @dermesut https://github.com/lllyasviel/stable-diffusion-webui-forge/issues/1421 Enables `sd_model_checkpoint` parameter for API payloads * Omit redundant .save() --- modules/api/api.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/modules/api/api.py b/modules/api/api.py index 6afb1c53..c40b4966 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -19,6 +19,7 @@ from secrets import compare_digest import modules.shared as shared from modules import sd_samplers, deepbooru, images, scripts, ui, postprocessing, errors, restart, shared_items, script_callbacks, infotext_utils, sd_models, sd_schedulers from modules.api import models +from modules_forge import main_entry from modules.shared import opts from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img, process_images from modules.textual_inversion.textual_inversion import create_embedding @@ -690,7 +691,9 @@ class Api: for k, v in req.items(): shared.opts.set(k, v, is_api=True) - shared.opts.save(shared.config_filename) + main_entry.checkpoint_change(checkpoint_name) + # shared.opts.save(shared.config_filename) --- applied in checkpoint_change() + return def get_cmd_flags(self):