fix aesthetic gradients doing nothing after loading a different model (0df94d3f) · Commits · github_fork / Stable Diffusion Webui

modules/sd_models.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -236,12 +236,11 @@ def load_model(checkpoint_info=None):
		sd_model.to(shared.device)

		sd_hijack.model_hijack.hijack(sd_model)
		script_callbacks.model_loaded_callback(sd_model)

		sd_model.eval()
		shared.sd_model = sd_model

		script_callbacks.model_loaded_callback(sd_model)

		print(f"Model loaded.")
		return sd_model

		@@ -268,6 +267,7 @@ def reload_model_weights(sd_model, info=None):
		load_model_weights(sd_model, checkpoint_info)

		sd_hijack.model_hijack.hijack(sd_model)
		script_callbacks.model_loaded_callback(sd_model)

		if not shared.cmd_opts.lowvram and not shared.cmd_opts.medvram:
		sd_model.to(devices.device)