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 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)