From af7cde4c80798cdf46af67d8387c6960bfa5a269 Mon Sep 17 00:00:00 2001 From: testIgnor <115854784+testIgnor@users.noreply.github.com> Date: Wed, 29 Jan 2025 15:32:52 -0600 Subject: [PATCH 1/2] Expose unprompted_seed and sd_seed --- scripts/unprompted.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/unprompted.py b/scripts/unprompted.py index f15c390..d6807c0 100644 --- a/scripts/unprompted.py +++ b/scripts/unprompted.py @@ -1085,6 +1085,8 @@ def process(self, p, is_enabled=True, unprompted_seed=-1, match_main_seed=True, Unprompted.shortcode_user_vars["sd_model"] = opts.data["sd_model_checkpoint"] Unprompted.shortcode_user_vars["sd_base"] = "none" Unprompted.shortcode_user_vars["sd_res"] = 1024 + Unprompted.shortcode_user_vars["unprompted_seed"] = unprompted_seed + Unprompted.shortcode_user_vars["sd_seed"] = p.seed if sd_models.model_data.sd_model: Unprompted.update_stable_diffusion_architecture_vars(sd_models.model_data.sd_model) From bc8f33fc42f723bbc6455fd4c233e1c2f3df9f10 Mon Sep 17 00:00:00 2001 From: testIgnor <115854784+testIgnor@users.noreply.github.com> Date: Fri, 31 Jan 2025 22:45:53 -0600 Subject: [PATCH 2/2] Fix memory leak when switching models --- lib_unprompted/shared.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib_unprompted/shared.py b/lib_unprompted/shared.py index e49ae68..03309ed 100644 --- a/lib_unprompted/shared.py +++ b/lib_unprompted/shared.py @@ -653,7 +653,7 @@ def update_stable_diffusion_vars(self, this_p): elif att == "sd_model" and self.shortcode_user_vars[att] != self.original_model and isinstance(self.shortcode_user_vars[att], str): info = sd_models.get_closet_checkpoint_match(self.shortcode_user_vars["sd_model"]) if info: - new_model = sd_models.load_model(info, None) #, None + new_model = sd_models.reload_model_weights(info=info) self.update_stable_diffusion_architecture_vars(new_model) elif att == "sd_vae": from modules import sd_vae