diff options
| author | MrCheeze <fishycheeze@yahoo.ca> | 2022-10-22 16:59:21 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-22 17:14:18 +0000 |
| commit | 0df94d3fcf9d1fc47c4d39039352a3d5b3380c1f (patch) | |
| tree | af1d0dcba71fec80e71116405a419ab14db3095e | |
| parent | 324c7c732dd9afc3d4c397c354797ae5d655b514 (diff) | |
| download | stable-diffusion-webui-gfx803-0df94d3fcf9d1fc47c4d39039352a3d5b3380c1f.tar.gz stable-diffusion-webui-gfx803-0df94d3fcf9d1fc47c4d39039352a3d5b3380c1f.tar.bz2 stable-diffusion-webui-gfx803-0df94d3fcf9d1fc47c4d39039352a3d5b3380c1f.zip | |
fix aesthetic gradients doing nothing after loading a different model
| -rw-r--r-- | modules/sd_models.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index f9b3063d..49dc3238 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -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)
|