diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-22 17:15:12 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-22 17:15:12 +0000 |
| commit | 321bacc6a9eaf4a25f31279f288fa752be507a20 (patch) | |
| tree | 80d1cec4253a5e7f91dcfc93e0ffeb5b874e1496 | |
| parent | 0df94d3fcf9d1fc47c4d39039352a3d5b3380c1f (diff) | |
| download | stable-diffusion-webui-gfx803-321bacc6a9eaf4a25f31279f288fa752be507a20.tar.gz stable-diffusion-webui-gfx803-321bacc6a9eaf4a25f31279f288fa752be507a20.tar.bz2 stable-diffusion-webui-gfx803-321bacc6a9eaf4a25f31279f288fa752be507a20.zip | |
call model_loaded_callback after setting shared.sd_model in case scripts refer to it using that
| -rw-r--r-- | modules/sd_models.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 49dc3238..e697bb72 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -236,11 +236,12 @@ 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
|