call model_loaded_callback after setting shared.sd_model in case scripts refer to it using that

3 years ago · 321bacc6a9
parent 0df94d3fcf
commit 321bacc6a9
1 changed files with 2 additions and 1 deletions
--- 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