apply unet overrides after switching model

2023-08-07 08:16:20 +03:00 · 2023-08-07 08:16:20 +03:00 · 6e7828e1d2
parent c96e4750d8
commit 6e7828e1d2
1 changed files with 1 additions and 0 deletions
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@ -699,6 +699,7 @@ def reload_model_weights(sd_model=None, info=None):
    print(f"Weights loaded in {timer.summary()}.")

    model_data.set_sd_model(sd_model)
+    sd_unet.apply_unet()

    return sd_model