Commit 321bacc6 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

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

parent 0df94d3f

modules/sd_models.py

+2 −1
Original line number Diff line number Diff line
@@ -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

CRA Git | Maintained and supported by SUSTech CRA and CCSE