Commit 8f96f928 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

call script callbacks for reloaded model after loading embeddings

parent 2d5a5076
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -324,12 +324,12 @@ def load_model(checkpoint_info=None):
    sd_model.eval()
    shared.sd_model = sd_model

    sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings(force_reload=True)  # Reload embeddings after model load as they may or may not fit the model

    script_callbacks.model_loaded_callback(sd_model)

    print("Model loaded.")

    sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings(force_reload = True) # Reload embeddings after model load as they may or may not fit the model

    return sd_model