Commit 3f828206 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

apply unet overrides after switching model

parent 6c7b6ecb
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -715,6 +715,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